Spaces:

keffy
/

hv_model

Sleeping

keffy commited on Oct 15

Commit

578b5c8

•

1 Parent(s): 7ba059b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import numpy as np
 from ultralytics import YOLO
 # Load the YOLOv8 model
-model = YOLO("yolov8n.pt")  # Ensure this file is in the same directory as app.py on Hugging Face
 # Define the inference function
 def predict(image):

 from ultralytics import YOLO
 # Load the YOLOv8 model
+model = YOLO("best.pt")  # Ensure this file is in the same directory as app.py on Hugging Face
 # Define the inference function
 def predict(image):