Update app.py

app.py

@@ -92,7 +92,10 @@ def reconstruct_image(img):
     out = decoder(z_q)
     out_img = np.clip(out[0].numpy() * 255, 0, 255).astype(np.uint8)
 
-    return out_img
+    out_img_pil = Image.fromarray(out_img)
+    out_img_upscaled = out_img_pil.resize((512, 512), Image.BICUBIC)
+
+    return out_img_upscaled
 
 # -------------------------
 # Gradio UI