TheKnight115 commited on
Commit
ad414f5
·
verified ·
1 Parent(s): e1fa062

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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")