Update app.py

app.py

@@ -67,7 +67,7 @@ def prompt_to_layout(user_prompt, top_p, top_k, fpath=None):
     model_prompt = '[User prompt] {} [Layout]'.format(user_prompt)
     print(model_prompt)
     input_ids = tokenizer(model_prompt, return_tensors='pt')
-    output = finetuned.generate(**input_ids, do_sample=True, top_p=top_p, top_k=top_k, max_length=300)
+    output = finetuned.generate(**input_ids, do_sample=True, top_p=top_p, top_k=top_k, eos_token_id=50256, max_length=400)
     output = tokenizer.batch_decode(output, skip_special_tokens=True)
     print(output)
     
@@ -135,7 +135,7 @@ def prompt_with_mutation(user_prompt, top_p, top_k, mut_rate, fpath=None):
     rooms = ', '.join(rooms)# + ', '
     new_prompt = '[User prompt] {} [Layout] {}'.format(user_prompt, rooms)
     input_ids = tokenizer(new_prompt, return_tensors='pt')
-    mut_output = finetuned.generate(**input_ids, do_sample=True, top_p=top_p, top_k=top_k, max_length=400)
+    mut_output = finetuned.generate(**input_ids, do_sample=True, top_p=top_p, top_k=top_k, eos_token_id=50256, max_length=400)
     mut_output = tokenizer.batch_decode(mut_output, skip_special_tokens=True)
     mut_im = mut_txt2layout(mut_output)