Fix bugs

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from PIL import Image
 import random
-from sahi.utils.yolov8
+#from sahi.utils.yolov8
 from sahi import AutoDetectionModel
 from utils import sahi_yolov8m_inference
 import sahi.utils.file
@@ -14,11 +14,11 @@ from streamlit_image_comparison import image_comparison
 #MMDET_YOLOX_TINY_CONFIG_URL = "https://huggingface.co/fcakyon/mmdet-yolox-tiny/raw/main/yolox_tiny_8x8_300e_coco.py"
 #MMDET_YOLOX_TINY_CONFIG_PATH = "config.py"
 
-#YOLOV8M_MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8m.pt"
-#YOLOV8M_MODEL_PATH = "tests/data/models/yolov8/yolov8m.pt"
+YOLOV8M_MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8m.pt"
+YOLOV8M_MODEL_PATH = "tests/data/models/yolov8/yolov8m.pt"
 
 
-YOLOV8M_MODEL_PATH = 'models/yolov8m.pt'
+#YOLOV8M_MODEL_PATH = 'models/yolov8m.pt'
 
 
 IMAGE_TO_URL = {
@@ -46,6 +46,10 @@ def download_comparison_images():
 @st.cache_data(show_spinner=False)
 def get_model():
     
+    sahi.utils.file.download_from_url(
+        YOLOV8M_MODEL_URL,
+        YOLOV8M_MODEL_PATH,
+    )
     #sahi.utils.file.download_from_url(
     #    MMDET_YOLOX_TINY_MODEL_URL,
     #    MMDET_YOLOX_TINY_MODEL_PATH,
@@ -55,7 +59,7 @@ def get_model():
     #    MMDET_YOLOX_TINY_CONFIG_PATH,
     #)
     
-    sahi.utils.yolov8.download_yolov8m_model(destination_path = YOLOV8M_MODEL_PATH)
+    #sahi.utils.yolov8.download_yolov8m_model(destination_path = YOLOV8M_MODEL_PATH)
 
     detection_model = AutoDetectionModel.from_pretrained(
         model_type='yolov8',

requirements.txt

@@ -3,8 +3,8 @@
 torch==1.12.1+cpu
 torchvision==0.13.1+cpu
 sahi==0.11.14
-mmdet==2.25.2
+#mmdet==2.25.2
 mmcv-full==1.6.1
 streamlit-image-comparison==0.0.4
 streamlit==1.22.0
-ultralyticsplus==0.0.14
+ultralyticsplus==0.0.28