Update app.py

app.py

@@ -7,12 +7,12 @@ import requests
 
 # Load the model and processor
 processor = AutoImageProcessor.from_pretrained("caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr")
-model = Swin2SRForImageSuperResolution.from_pretrained("caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr").to('cuda')
+model = Swin2SRForImageSuperResolution.from_pretrained("caidas/swin2SR-realworld-sr-x4-64-bsrgan-psnr").to('cpu')
 
 # Define the function for super-resolution
 def super_resolve(image):
     # Preprocess the input image
-    inputs = processor(images=image, return_tensors="pt").to('cuda')
+    inputs = processor(images=image, return_tensors="pt").to('cpu')
 
     # Perform super-resolution
     with torch.no_grad():