Spaces:

jameslahm
/

YOLOv10

Running on Zero

jameslahm commited on Jun 3, 2024

Commit

794b1a6

verified ·

1 Parent(s): 810cd5d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import PIL.Image as Image
 import gradio as gr
-from ultralytics import ASSETS, YOLOv10
 def predict_image(img, model_id, image_size, conf_threshold):
     model = YOLOv10.from_pretrained(f'jameslahm/{model_id}')

 import PIL.Image as Image
 import gradio as gr
+from ultralytics import YOLOv10
 def predict_image(img, model_id, image_size, conf_threshold):
     model = YOLOv10.from_pretrained(f'jameslahm/{model_id}')