Update app.py

app.py

@@ -1,7 +1,7 @@
 # Import necessary libraries
 from PIL import Image
-import torch
 import numpy as np
+import torch
 from transformers import AutoImageProcessor, Swin2SRForImageSuperResolution
 import gradio as gr  # Import Gradio for creating the interface
 
@@ -35,7 +35,6 @@ def main(image, save_as_jpg=True):
     # Check if GPU is available and set the device accordingly
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
-    regular_model = "caidas/swin2SR-classical-sr-x4-64"
     realworld_model = "caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr"
 
     # Load the Swin2SR model and processor for 4x upscaling