Tennineee commited on
Commit
3b2067a
·
verified ·
1 Parent(s): 68b8ca8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="PDFNet-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
 
 
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