Update app.py

app.py

@@ -1,5 +1,5 @@
 from diffusers import AutoPipelineForText2Image, StableDiffusionImg2ImgPipeline
-from PIL import Image
+import torch.nn.functional as F
 import gradio as gr
 import random
 import torch
@@ -18,17 +18,17 @@ css = """
 def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
     if seed < 0:
         seed = random.randint(1,999999)
-    image = txt2img(
+    latent = txt2img(
         prompt,
         num_inference_steps=1,
         guidance_scale=0.0,
         generator=torch.manual_seed(seed),
         output_type="latent"
     ).images[0]
-    upscaled_image = image.resize((4,128,128), 1)
+    upscaled_latent = F.interpolate(latent, size=128)
     final_image = img2img(
         prompt,
-        upscaled_image,
+        upscaled_latent,
         num_inference_steps=int(samp_steps*2),
         guidance_scale=5,
         strength=0.5,