Update app.py

app.py

@@ -7,7 +7,7 @@ import spaces
 
 device = "cuda" 
 
-pipe = DiffusionPipeline.from_pretrained("aipicasso/emi-2")
+pipe = DiffusionPipeline.from_pretrained("aipicasso/emi-2",torch_dtype=torch.bfloat16)
 pipe = pipe.to(device)
 
 
@@ -34,7 +34,7 @@ def infer(seed, randomize_seed, width, height, guidance_scale, num_inference_ste
         generator = generator
     ).images[0] 
     
-    return image
+    return image, prompt
 
 css="""
 #col-container {
@@ -101,13 +101,13 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=30,
                     step=1,
-                    value=25,
+                    value=20,
                 )
 
     run_button.click(
         fn = infer,
         inputs = [seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
-        outputs = [result]
+        outputs = [result,"textbox"]
     )
 
 demo.queue().launch()