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 |
|