Update app.py

app.py

@@ -46,7 +46,7 @@ def inference(img, task, run_on):
     load_arch = run_path('restormer_arch.py')
     model = load_arch['Restormer'](**parameters)
     
-    checkpoint = torch.load('deshabby.pt')
+    checkpoint = torch.load('deshabby.pth')
     model.load_state_dict(checkpoint['params'])
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  
     model = model.to(device)