Update app.py

app.py

@@ -25,7 +25,8 @@ def upscale_image(model, input_image):
     if model == "SD 2.0 2x Latent Upscaler":
         upscaled_image = sd_2_0_2x(prompt="", image=input_image, num_inference_steps=5).images[0]
     else:
-        upscaled_image = sd_2_1_4x(prompt="", image=input_image, num_inference_steps=5).images[0]
+        low_res_img = input_image.resize((128, 128))
+        upscaled_image = sd_2_1_4x(prompt="", image=low_res_img, num_inference_steps=5).images[0]
     
     # Convert the upscaled tensor back to a PIL image