linhcuem commited on
Commit
c6bd81c
1 Parent(s): b9568f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -5,13 +5,12 @@ import yolov5
5
  from PIL import Image
6
  from huggingface_hub import hf_hub_download
7
  import os
8
- from ultralyticsplus import YOLO
9
 
10
 
11
  app_title = "Detect defects in bird nest jar"
12
  models_ids = ['linhcuem/defects_bird_nest_jar_yolov5m6','linhcuem/defects_bird_nest_jar_yolov5x']
13
- # DEFAULT_DET_MODEL_ID_yolov5 = 'linhcuem/defects_bird_nest_jar_yolov5m6'
14
- # DEFAULT_DET_MODEL_ID_yolov8 = 'linhcuem/defects_bird_nest_jar_yolov8'
15
 
16
  current_model_id = models_ids[-1]
17
  model = yolov5.load(current_model_id)
 
5
  from PIL import Image
6
  from huggingface_hub import hf_hub_download
7
  import os
8
+
9
 
10
 
11
  app_title = "Detect defects in bird nest jar"
12
  models_ids = ['linhcuem/defects_bird_nest_jar_yolov5m6','linhcuem/defects_bird_nest_jar_yolov5x']
13
+
 
14
 
15
  current_model_id = models_ids[-1]
16
  model = yolov5.load(current_model_id)