Update app.py

app.py

@@ -23,16 +23,18 @@ def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
         num_inference_steps=1,
         guidance_scale=0.0,
         generator=torch.manual_seed(seed),
-    ).images
-    upscaled_image = image[0].resize((1024,1024), 1)
+        output_type="latent"
+    ).images[0]
+    upscaled_image = image.resize((1024,1024), 1)
     final_image = img2img(
         prompt,
+        upscaled_image,
         num_inference_steps=samp_steps,
         guidance_scale=5,
         strength=0.5,
         generator=torch.manual_seed(seed),
-    ).images
-    return gr.update(value = (final_image[0], f"Image: {prompt}")), seed
+    ).images[0]
+    return gr.update(value = (final_image, f"Image: {prompt}")), seed
         
 def set_base_models():
     txt2img = AutoPipelineForText2Image.from_pretrained(