Update app.py

app.py

@@ -41,10 +41,10 @@ def inference(img, task, run_on):
         img = img.resize((width,height), Image.ANTIALIAS)
       
     #parameters = {'inp_channels':3, 'out_channels':3, 'dim':48, 'num_blocks':[4,6,6,8], 'num_refinement_blocks':4, 'heads':[1,2,4,8], 'ffn_expansion_factor':2.66, 'bias':False, 'LayerNorm_type':'WithBias', 'dual_pixel_task':False}
-    model = torch.load('deshabby.pt')
-    model.load_state_dict(model['params'])
+    #checkpoint = torch.load('deshabby.pt')
+    #model.load_state_dict(checkpoint['params'])
     #model.eval()
-    #model = torch.jit.load('deshabby.pt') 
+    model = torch.jit.load('deshabby.pt') 
     
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')  
     model = model.to(device)