Update app.py

app.py

@@ -4,10 +4,11 @@ import gradio as gr
 from ultralytics import YOLO
 from transformers import AutoImageProcessor, AutoModelForDepthEstimation
 from PIL import Image
+import torch  # Added torch import for depth estimation
 
 # Load YOLO model for tree detection
-# Replace "path_to_your_yolo_model.pt" with your model path (e.g., local or Hugging Face Hub)
-yolo_model = YOLO("./data/best.pt")  # Update with your YOLO model path
+# Replace with your model path (local or Hugging Face Hub)
+yolo_model = YOLO("./data/best.pt") # Update with your YOLO model path
 
 # Load depth estimation model and processor from Hugging Face
 processor = AutoImageProcessor.from_pretrained("Intel/dpt-large")
@@ -106,5 +107,5 @@ iface = gr.Interface(
     description="Upload an image to detect trees and estimate their heights in centimeters. Adjust camera parameters and depth scale as needed."
 )
 
-# Launch the interface
-iface.launch()
+# Launch the interface locally
+iface.launch(server_name="0.0.0.0", server_port=7860)