keremberke commited on
Commit
6982d7b
1 Parent(s): 87f77e0

upload space files

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ article = f"<p style='text-align: center'> <a href='https://huggingface.co/{mode
13
  current_model_id = models_ids[-1]
14
  model = yolov5.load(current_model_id)
15
 
16
- if test_images:
17
  image_filenames = os.listdir(test_images)
18
  examples = [
19
  [Path(test_images) / image_filename, 0.25, models_ids[-1]]
 
13
  current_model_id = models_ids[-1]
14
  model = yolov5.load(current_model_id)
15
 
16
+ if "test_images":
17
  image_filenames = os.listdir(test_images)
18
  examples = [
19
  [Path(test_images) / image_filename, 0.25, models_ids[-1]]