Spaces:

to-be
/

Scanned_document_denoise_reconstruct

Running

to-be commited on Sep 24, 2023

Commit

8cbfac3

1 Parent(s): c24ebdd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

         img = img.resize((width,height), Image.ANTIALIAS)
+    model = torch.jit.load('deshabby.pt')
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     model = model.to(device)
     model.eval()