Update app.py

app.py

@@ -32,7 +32,7 @@ def inference(img):
       
     # 'Downsampled Image'
     ####  Resize the longer edge of the input image
-      max_res = 512
+      max_res = 200
       width, height = img.size
       if max(width,height) > max_res:
         scale = max_res /max(width,height)
@@ -45,7 +45,7 @@ def inference(img):
     load_arch = run_path('restormer_arch.py')
     model = load_arch['Restormer'](**parameters)
     
-    checkpoint = torch.load('deshabby.pth')
+    checkpoint = torch.load('net_g_92000.pth')
     model.load_state_dict(checkpoint['params'])
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  
     model = model.to(device)