Update app.py

app.py

@@ -1,19 +1,17 @@
 import gradio as gr
 import mlflow
-import dagshub
-from ultralytics import YOLO
-from PIL import Image
-import cv2
-import numpy as np
 import os
 import time
 import tempfile
 import shutil
+from ultralytics import YOLO
+from PIL import Image
+import cv2
+import numpy as np
 
 # ==============================
-# MLflow / DagsHub Configuration using Secrets
+# MLflow Configuration using Secrets ONLY
 # ==============================
-# Secrets من HF Spaces
 tracking_uri = os.getenv("MLFLOW_TRACKING_URI")
 username = os.getenv("MLFLOW_TRACKING_USERNAME")
 password = os.getenv("MLFLOW_TRACKING_PASSWORD")
@@ -25,18 +23,14 @@ os.environ["MLFLOW_TRACKING_URI"] = tracking_uri
 os.environ["MLFLOW_TRACKING_USERNAME"] = username
 os.environ["MLFLOW_TRACKING_PASSWORD"] = password
 
-# تفعيل DagsHub
-dagshub.init(repo_owner="Mosensei7", repo_name="AutonomousVehiclesDetectionDEPI", mlflow=True)
-
 # إنشاء experiment
 mlflow.set_experiment("YOLOv12s_Inference_Logs")
-
-print("MLflow connected successfully to DagsHub!")
+print("✅ MLflow configured successfully using secrets!")
 
 # ==============================
 # Load Model
 # ==============================
-model = YOLO("Yolo12s.pt")  # اسم الـ repo بتاعك
+model = YOLO("Mosensei7/yolov12s-egyptian-autonomous-vehicles/best.pt")  # اسم الـ repo بتاعك
 
 # ==============================
 # Inference with Full MLflow Tracking
@@ -128,7 +122,7 @@ h1 { text-align: center; color: #00ffff; text-shadow: 0 0 20px #00ffff; font-siz
 button { background: linear-gradient(45deg, #ff00ff, #00ffff); border: none; border-radius: 20px; height: 60px; font-size: 1.3em; box-shadow: 0 0 25px #00ffff; }
 """
 
-with gr.Blocks(css=css, theme=gr.themes.Dark()) as demo:
+with gr.Blocks(css=css, theme=gr.themes.Monochrome()) as demo:  # ✅ FIXED THEME HERE
     gr.Markdown("""
     # 🚀 YOLOv12s - كشف المركبات في الشوارع المصرية 🇪🇬
     **كل عملية كشف بتترفع تلقائيًا على DagsHub MLflow مع الـ input والـ output والإحصائيات**