Update app.py

app.py

@@ -41,7 +41,7 @@ def mingle_prompts(first_prompt, second_prompt):
         mixed_embeddings = (text_embeddings1 * mix_factor + text_embeddings2 * (1 - mix_factor))
 
         # Generate!
-        steps = 30
+        steps = 20
         guidence_scale = 8.0
         img = mingle_model.generate_with_embs(mixed_embeddings, rand_generator, num_inference_steps=steps,
                                  guidance_scale=guidence_scale)
@@ -66,7 +66,7 @@ with gr.Blocks() as demo:
                  ['A mouse', 'A leopard']], [first_prompt, second_prompt])
 
     gr.Markdown("# Output Results")
-    output = gr.Image()
+    output = gr.Image(shape=(512,512))
 
     greet_btn.click(fn=mingle_prompts, inputs=[first_prompt, second_prompt], outputs=[output])