keremberke commited on
Commit
40bbc83
1 Parent(s): 5293ae3

upload space files

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,10 +13,10 @@ 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 space_content/test_images:
17
- image_filenames = os.listdir(space_content/test_images)
18
  examples = [
19
- [Path(space_content/test_images) / image_filename, 0.25, models_ids[-1]]
20
  for image_filename in image_filenames
21
  ]
22
  else:
 
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]]
20
  for image_filename in image_filenames
21
  ]
22
  else: