Update app.py

app.py

@@ -65,11 +65,11 @@ def draw_polygons(polygons, colors, im_size=(256, 256), b_color="white", fpath=N
 def prompt_to_layout(user_prompt, top_p, top_k, fpath=None):
     
     model_prompt = '[User prompt] {} [Layout]'.format(user_prompt)
-    print(model_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, eos_token_id=50256, max_length=400)
     output = tokenizer.batch_decode(output, skip_special_tokens=True)
-    print(output)
+    #print(output)
     
     layout = output[0].lstrip().split('[User prompt]')[1].split('[Layout]')[1].split(', ')
     spaces = [txt.split(':')[0] for txt in layout]