Update app.py

app.py

@@ -25,7 +25,8 @@ def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
         generator=torch.manual_seed(seed),
         output_type="latent"
     ).images[0]
-    upscaled_latent = F.interpolate(latent, (128,128))
+    latent = latent.unsqueeze(0)
+    upscaled_latent = F.interpolate(latent, scale_factor=2)
     final_image = img2img(
         prompt,
         upscaled_latent,