abdulrafay9 commited on
Commit
8366f47
·
verified ·
1 Parent(s): b20232a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ core_path = hf_hub_download(
23
  weights_path = hf_hub_download(
24
  repo_id=PRIVATE_DATASET_ID,
25
  repo_type="dataset",
26
- filename="best.pt",
27
  token=TOKEN,
28
  )
29
 
 
23
  weights_path = hf_hub_download(
24
  repo_id=PRIVATE_DATASET_ID,
25
  repo_type="dataset",
26
+ filename="model.pth",
27
  token=TOKEN,
28
  )
29