Update app.py

Changed the default when no model is selected to the 2x upscaler

app.py

@@ -21,12 +21,12 @@ def upscale_image(model, input_image):
     generator = torch.manual_seed(999999)
     input_image = Image.open(input_image).convert("RGB")
     # 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, num_inference_steps=5).images[0]
-    else:
+    if model == "SD 2.1 4x Upscaler":
         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]
-    # Return the upscaled image
+    else:
+        upscaled_image = sd_2_0_2x(prompt="", image=input_image, num_inference_steps=5).images[0]
+        # Return the upscaled image
     return upscaled_image
 
 # Define the Gradio interface