nagasurendra commited on
Commit
bc2a181
·
verified ·
1 Parent(s): 4f24d2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
 
8
  # Load YOLO model for tree detection
9
  # Replace "path_to_your_yolo_model.pt" with your model path (e.g., local or Hugging Face Hub)
10
- yolo_model = YOLO("path_to_your_yolo_model.pt") # Update with your YOLO model path
11
 
12
  # Load depth estimation model and processor from Hugging Face
13
  processor = AutoImageProcessor.from_pretrained("Intel/dpt-large")
 
7
 
8
  # Load YOLO model for tree detection
9
  # Replace "path_to_your_yolo_model.pt" with your model path (e.g., local or Hugging Face Hub)
10
+ yolo_model = YOLO("./data/best.pt") # Update with your YOLO model path
11
 
12
  # Load depth estimation model and processor from Hugging Face
13
  processor = AutoImageProcessor.from_pretrained("Intel/dpt-large")