delphiclinic commited on
Commit
f760cbf
·
verified ·
1 Parent(s): 152ea8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
 
7
  ## Where to store flagged data
8
  HF_TOKEN = os.getenv('HF_TOKEN')
9
- hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "flaggedImages/images")
10
 
11
  # Load your trained model
12
  model = YOLO('best.pt')
 
6
 
7
  ## Where to store flagged data
8
  HF_TOKEN = os.getenv('HF_TOKEN')
9
+ hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "flaggedImages")
10
 
11
  # Load your trained model
12
  model = YOLO('best.pt')