Update app.py

app.py

@@ -41,8 +41,8 @@ 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}
-    checkpoint = torch.load('deshabby.pt')
-    model.load_state_dict(checkpoint['params'])
+    model = torch.load('deshabby.pt')
+    model.load_state_dict(model['params'])
     #model.eval()
     #model = torch.jit.load('deshabby.pt')