Update app.py

app.py

@@ -60,13 +60,24 @@
 
 
 
+import os
+import requests
 from ultralytics import YOLO
 import pandas as pd
 from PIL import Image, ImageDraw
 import gradio as gr
 
+# تحميل الموديل من Hugging Face لو مش موجود
+model_path = "best.pt"
+model_url = "https://huggingface.co/AlaaElsayed/yolospace/resolve/main/best.pt"
+
+if not os.path.exists(model_path):
+    r = requests.get(model_url)
+    with open(model_path, "wb") as f:
+        f.write(r.content)
+
 # تحميل الموديل
-model = YOLO("best.pt")
+model = YOLO(model_path)
 
 # تحميل بيانات التغذية
 food_df = pd.read_csv("food_cleaned.csv")
@@ -89,7 +100,7 @@ def detect(image):
     boxes = result.boxes
     names = model.names
 
-    img = Image.fromarray(result.plot())  # الصورة عليها البوكسات
+    img = Image.fromarray(result.plot())  # الصورة مع البوكسات
     draw = ImageDraw.Draw(img)
 
     for box in boxes:
@@ -111,3 +122,4 @@ demo = gr.Interface(
 )
 
 demo.launch()
+