Update streamlit_app.py

streamlit_app.py

@@ -8,6 +8,11 @@ import os
 from datetime import datetime
 import streamlit as st
 import numpy as np
+from ultralytics import YOLOvv8
+
+MODEL_PATH = YOLOvv8.from_pretrained("Ultralytics/YOLOv8")
+#source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
+#model.predict(source=source, save=True)
 
 # Initialize models
 #plate_model = YOLO('yolov8n.pt')
@@ -18,10 +23,10 @@ from ultralytics.utils.downloads import attempt_download_asset
 if not os.path.exists('yolov8n.pt'):
     attempt_download_asset('yolov8n.pt')
 
-MODEL_PATH = 'yolov8n.pt'
-if not os.path.exists(MODEL_PATH):
-    from urllib.request import urlretrieve
-    urlretrieve("https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt", MODEL_PATH)
+#MODEL_PATH = 'yolov8n.pt'
+#if not os.path.exists(MODEL_PATH):
+#    from urllib.request import urlretrieve
+#    urlretrieve("https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt", MODEL_PATH)
 
 plate_model = YOLO(MODEL_PATH)