Update app.py

app.py

@@ -22,8 +22,7 @@ if not os.path.exists(MODEL_PATH):
 # Define the Real-ESRGAN class
 class RealESRGAN:
     def __init__(self, model_path, device):
-        # Correctly load the model
-        self.model = torch.hub.load('xinntao/Real-ESRGAN', 'real_esrgan', model_path, device=device)
+        self.model = torch.hub.load('xinntao/Real-ESRGAN', 'real_esrgan', model_path, device=device, trust_repo=True)
         self.model.eval()  # Set the model to eval mode
         self.device = device  # Store the device