Update app.py

app.py

@@ -1,17 +1,24 @@
 import gradio as gr
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_download
 from PIL import Image
 
-# Load your YOLO model (it must be in the same repo or use snapshot from Hugging Face Hub)
-model = YOLO("Safi029/ABD-model/blob/main/ABD.pt")
+# ✅ Download your model from Hugging Face Hub
+model_path = hf_hub_download(repo_id="Safi029/ABD-model", filename="ABD.pt")
+model = YOLO(model_path)
 
 def detect_structure(image):
     results = model(image)
-    return results[0].plot()  # Returns image with boxes
+    annotated_img = results[0].plot()
+    return Image.fromarray(annotated_img)
 
-gr.Interface(
+demo = gr.Interface(
     fn=detect_structure,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Image(type="numpy"),
-    title="Molecular Structure Detection"
-).launch()
+    outputs=gr.Image(type="pil"),
+    title="YOLO Molecular Structure Reader",
+    description="Upload a molecular structure image. The YOLO model will detect atoms and bonds."
+)
+
+demo.launch()
+