rgp commited on
Commit
2b7611c
1 Parent(s): 7008721

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import hf_hub_download
4
  from PIL import Image
5
 
6
  REPO_ID = "rgp/yolov5-street-view-detection-grayscale"
7
- FILENAME = "best.pt"
8
 
9
  yolov5_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
10
 
 
4
  from PIL import Image
5
 
6
  REPO_ID = "rgp/yolov5-street-view-detection-grayscale"
7
+ FILENAME = "best_gray.pt"
8
 
9
  yolov5_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
10