ProConceptTech commited on
Commit
b6b3664
·
verified ·
1 Parent(s): 6d484ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,9 +1,13 @@
1
  import gradio as gr
2
  from ultralytics import YOLO
3
- from PIL import Image
4
- import numpy as np
5
 
6
- model = YOLO("best.pt")
 
 
 
 
 
7
 
8
  def detect(image):
9
  results = model(image)
 
1
  import gradio as gr
2
  from ultralytics import YOLO
3
+ from huggingface_hub import hf_hub_download
 
4
 
5
+ model_path = hf_hub_download(
6
+ repo_id="ProConceptTech/jambazisight",
7
+ filename="best.pt"
8
+ )
9
+
10
+ model = YOLO(model_path)
11
 
12
  def detect(image):
13
  results = model(image)