Update app.py

app.py

@@ -20,18 +20,18 @@ pipe = pipe.to(device)
 
 
 # Funzione per generare un'immagine
-def generate_image(prompt, neg_prompt, seed, steps):
+def generate_image(prompt, neg_prompt, width=resolution[0], height=resolution[1], seed, steps):
     generator = torch.manual_seed(seed)
     start_time = time.time()
-    image = pipe(prompt, height=resolution[1], width=resolution[0], num_inference_steps=steps, guidance_scale=guidance_scale, generator=generator, negative_prompt=neg_prompt).images[0]
+    image = pipe(prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=guidance_scale, generator=generator, negative_prompt=neg_prompt).images[0]
     end_time = time.time()    
     elapsed_time = end_time - start_time
     return image, elapsed_time
 
 demo = gr.Interface(
     fn=generate_image,
-    inputs=["text","text", "slider", "slider"],
-    outputs=[gr.Image(),"textbox"],
+    inputs=["text","text","number","number", "slider", "slider"],
+    outputs=[gr.Image(),gr.Textbox(label="Elapsed time (seconds)")],
 )
 
 demo.launch()