Hugging Face's logo

Spaces:
poudel
/
Yolov8_Cargo_counter
Sleeping

App Files Community
poudel commited on
Commit
6607502
·
verified ·
1 Parent(s): 4404024

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from ultralytics import YOLO
3
 
4
  # Load YOLOv11 models from Hugging Face Hub
5
- best_model = YOLO('https://huggingface.co/poudel/yolov8-cargo-package-counter/blob/main/1/best.pt')
6
- last_model = YOLO('https://huggingface.co/poudel/yolov8-cargo-package-counter/blob/main/1/last.pt')
7
 
8
  # Function to detect and count packages using the selected model
9
  def count_packages(image, model_choice):
 
2
  from ultralytics import YOLO
3
 
4
  # Load YOLOv11 models from Hugging Face Hub
5
+ best_model = YOLO('https://huggingface.co/poudel/yolov8-cargo-package-counter/blob/main/best.pt')
6
+ last_model = YOLO('https://huggingface.co/poudel/yolov8-cargo-package-counter/blob/main/last.pt')
7
 
8
  # Function to detect and count packages using the selected model
9
  def count_packages(image, model_choice):