Update app.py

app.py

@@ -41,7 +41,7 @@ def inference(img, task, run_on):
         img = img.resize((width,height), Image.ANTIALIAS)
       
 
-    model = torch.jit.load('deraining.pt')    
+    model = torch.jit.load('deshabby.pth')    
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  
     model = model.to(device)
     model.eval()