Update app.py

app.py

@@ -24,7 +24,7 @@ def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
         guidance_scale=0.0,
         generator=torch.manual_seed(seed),
     ).images
-    upscaled_image = image.resize((1024,1024), 1)
+    upscaled_image = image[0].resize((1024,1024), 1)
     final_image = img2img(
         prompt,
         num_inference_steps=samp_steps,
@@ -32,7 +32,7 @@ def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
         strength=0.5,
         generator=torch.manual_seed(seed),
     ).images
-    return gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(final_images)]), seed
+    return gr.update(value = (final_image[0], f"Image: {prompt}"), seed
         
 def set_base_models():
     txt2img = AutoPipelineForText2Image.from_pretrained(