valentynliubchenko commited on
Commit
f122098
·
1 Parent(s): 02e3ba7

fixed path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import numpy as np
6
  model_options = ["xViewyolov8n_v8_100e.pt", "xViewyolov8s_v8_100e.pt", "xViewyolov8m_v8_100e.pt"]
7
  model_names = ["Nano", "Small", "Medium"]
8
  models = [YOLO(os.path.join("./saved_model", option)) for option in model_options]
9
- example_list = [["examples/" + example] for example in os.listdir("examples")]
10
 
11
  def process_image(input_image, model_name, conf):
12
  print(model_name)
 
6
  model_options = ["xViewyolov8n_v8_100e.pt", "xViewyolov8s_v8_100e.pt", "xViewyolov8m_v8_100e.pt"]
7
  model_names = ["Nano", "Small", "Medium"]
8
  models = [YOLO(os.path.join("./saved_model", option)) for option in model_options]
9
+ example_list = [["./examples/" + example] for example in os.listdir("examples")]
10
 
11
  def process_image(input_image, model_name, conf):
12
  print(model_name)