Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ device = torch.device('cpu')
|
|
| 44 |
parser = argparse.ArgumentParser('PDFNet Testing script', parents=[get_args_parser()])
|
| 45 |
args = parser.parse_args(args=[])
|
| 46 |
model,model_name = build_model(args)
|
| 47 |
-
model_path = hf_hub_download(repo_id="Tennineee/PDFNet-general",filename="
|
| 48 |
model.load_state_dict(torch.load(model_path,map_location='cpu'),strict=False)
|
| 49 |
model = model.to(device).eval()
|
| 50 |
|
|
|
|
| 44 |
parser = argparse.ArgumentParser('PDFNet Testing script', parents=[get_args_parser()])
|
| 45 |
args = parser.parse_args(args=[])
|
| 46 |
model,model_name = build_model(args)
|
| 47 |
+
model_path = hf_hub_download(repo_id="Tennineee/PDFNet-general",filename="PDF-Generally.pth", repo_type="model")
|
| 48 |
model.load_state_dict(torch.load(model_path,map_location='cpu'),strict=False)
|
| 49 |
model = model.to(device).eval()
|
| 50 |
|