Update app.py

app.py

@@ -45,6 +45,7 @@ parser = argparse.ArgumentParser('PDFNet Testing script', parents=[get_args_pars
 args = parser.parse_args(args=[])
 model,model_name = build_model(args)
 model_path = hf_hub_download(repo_id="Tennineee/PDFNet",filename="PDFNet_Best.pth", repo_type="model")
+model.load_state_dict(torch.load(model_path,map_location='cpu'),strict=False)
 model = model.to(device).eval()
 
 DAMV2_configs = {