Update app.py

app.py

@@ -1,12 +1,23 @@
 import gradio as gr
 from ultralytics import YOLO
 from PIL import Image
+from huggingface_hub import hf_hub_download
 import numpy as np
 import os
 
-# Load the YOLO model - the model will be loaded from the HF repo
-model_repo = "ashen007/document-structure-detection"
-model = YOLO(model_repo)
+# Load the YOLO model - use the proper format for Hugging Face repos
+try:
+    # Format for loading from Hugging Face: "hf://username/model_name"
+    model_repo = "hf://ashen007/document-structure-detection"
+    model = YOLO(model_repo)
+    print(f"Successfully loaded model from {model_repo}")
+    
+except Exception as e:
+    model_path = hf_hub_download(
+      repo_id="ashen007/document-structure-detection", 
+      filename="DSD-YOLOv8-v2.pt"
+    )
+    model = YOLO(model_path)
 
 # Define your class names - update this with your actual class names
 class_names = [
@@ -66,6 +77,6 @@ demo = gr.Interface(
     ]
 )
 
-# Launch the app
+# Launch the app - settings for Hugging Face Spaces
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=False)