Update app.py

app.py

@@ -3,14 +3,14 @@ import torch
 from ultralytics import YOLO
 import gradio as gr
 
-# Load the pre-trained YOLO model (assuming 'best.pt' is a YOLOv5 model)
-model = YOLO("./data/best.pt")
+# Load the pre-trained YOLOv8 model
+model = YOLO("./data/best.pt")  # Replace with path to your trained YOLOv8 model
 
 # Function to process video frames and count wine bottles
 def process_frame(frame):
     # Perform inference on the frame
     results = model(frame)
-    
+
     # Extract results
     detections = results.pandas().xywh[results.pandas().xywh['class'] == 0]  # Assuming '0' is the class for wine bottles