Spaces:
Runtime error
Runtime error
Daniel Cerda Escobar
commited on
Commit
·
74147bf
1
Parent(s):
857c128
Fix download model
Browse files
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
|
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',
|