Update app.py

app.py

@@ -23,9 +23,9 @@ def upscale_image(model, input_image):
     
     # Upscale the image using the selected model
     if model == "SD 2.0 2x Latent Upscaler":
-        upscaled_image = sd_2_0_2x(prompt="", image=input_image)
+        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)
+        upscaled_image = sd_2_1_4x(prompt="", image=input_image, num_inference_steps=5).images[0]
     
     # Convert the upscaled tensor back to a PIL image