Spaces:

newtechdevng
/

detect-api

Sleeping

newtechdevng commited on 15 days ago

Commit

6b530b5

verified ·

1 Parent(s): a6288da

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from PIL import Image
 # Load your model from your HF repo
 model = torch.load(
-    hf_hub_download(repo_id="newtechdevng/detect", filename="your_model_file.pt"),
     map_location="cpu"
 )
 model.eval()

 # Load your model from your HF repo
 model = torch.load(
+    hf_hub_download(repo_id="newtechdevng/detect", filename="best.pt"),
     map_location="cpu"
 )
 model.eval()