Update app.py to download YOLOv8 model using Hugging Face token and adjust description text. Clean up requirements.txt by removing unnecessary dependencies.

app.py

@@ -1,16 +1,21 @@
+import os
 import gradio as gr
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_download
 
-# Load YOLOv8 model from Hugging Face
-model = YOLO(
-    "https://huggingface.co/Lookingsoft-team/object_detection/resolve/main/yolov8n.pt"
+# Download private model using HF token from secret
+hf_token = os.environ.get("HF_TOKEN")
+model_path = hf_hub_download(
+    repo_id="Lookingsoft-team/object_detection", filename="yolov8n.pt", token=hf_token
 )
 
+# Load model
+model = YOLO(model_path)
+
 
 def detect_objects(image):
     results = model(image)
-    annotated_image = results[0].plot()  # Draw bounding boxes
-    return annotated_image
+    return results[0].plot()
 
 
 demo = gr.Interface(
@@ -18,7 +23,7 @@ demo = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=gr.Image(type="numpy"),
     title="Object Detection",
-    description="Upload an image for object detection!",
+    description="Upload an image to detect objects.",
 )
 
 if __name__ == "__main__":

requirements.txt

@@ -1,5 +1,3 @@
 gradio>=5.32.1
 ultralytics
-torch
-numpy
-Pillow 
+huggingface_hub