Daniel Cerda Escobar commited on
Commit
74147bf
·
1 Parent(s): 857c128

Fix download model

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -1,9 +1,7 @@
1
  import streamlit as st
2
  from PIL import Image
3
  import random
4
- from sahi.utils.yolov8 import (
5
- download_yolov8m_model,
6
- )
7
  from sahi import AutoDetectionModel
8
  from utils import sahi_yolov8m_inference
9
  import sahi.utils.file
@@ -57,7 +55,7 @@ def get_model():
57
  # MMDET_YOLOX_TINY_CONFIG_PATH,
58
  #)
59
 
60
- download_yolov8m_model(destination_path = YOLOV8M_MODEL_PATH)
61
 
62
  detection_model = AutoDetectionModel.from_pretrained(
63
  model_type='yolov8',
 
1
  import streamlit as st
2
  from PIL import Image
3
  import random
4
+ from sahi.utils.yolov8
 
 
5
  from sahi import AutoDetectionModel
6
  from utils import sahi_yolov8m_inference
7
  import sahi.utils.file
 
55
  # MMDET_YOLOX_TINY_CONFIG_PATH,
56
  #)
57
 
58
+ sahi.utils.yolov8.download_yolov8m_model(destination_path = YOLOV8M_MODEL_PATH)
59
 
60
  detection_model = AutoDetectionModel.from_pretrained(
61
  model_type='yolov8',