Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,9 @@ import streamlit as st
|
|
2 |
import cv2
|
3 |
import numpy as np
|
4 |
from ultralytics import YOLO
|
|
|
|
|
|
|
5 |
|
6 |
# Load the YOLO model
|
7 |
model = YOLO("TheKnight115/yolov8_Medium")
|
|
|
2 |
import cv2
|
3 |
import numpy as np
|
4 |
from ultralytics import YOLO
|
5 |
+
from huggingface_hub import hf_hub_download
|
6 |
+
|
7 |
+
model_file = hf_hub_download(repo_id="TheKnight115/Yolov8m", filename="yolov8_Medium.pt")
|
8 |
|
9 |
# Load the YOLO model
|
10 |
model = YOLO("TheKnight115/yolov8_Medium")
|