Attempting Model Fix

app.py

@@ -6,7 +6,7 @@ from ultralytics import YOLO
 import os
 
 MARKDOWN = """
-<h1 style='text-align: left'>YOLO Application Toolkit 🚀🌍</h1>
+<h1 style='text-align: left'>YOLO Application Toolkit 🚀</h1>
 <p>Welcome to the YOLO Application Toolkit! This demo highlights the powerful detection capabilities of various YOLO models pre-trained on different datasets. 🎉
 
 Easily detect objects in images and videos on the go. Perfect for quick experimentation and practical use. 🎉🔍</p>
@@ -26,16 +26,14 @@ Powered by Roboflow [Inference](https://github.com/roboflow/inference),
 [Supervision](https://github.com/roboflow/supervision) and [Ultralytics](https://github.com/ultralytics/ultralytics).🔥
 """
 
-MODEL_DIR = os.path.join(os.path.dirname(__file__), "models")
-
 # Load models dynamically
 MODELS = {
-    "YOLO11m (COCO128)": YOLO(os.path.join(MODEL_DIR, "yolo11m.pt")),
-    "American Sign Language (ASL) YOLO11m": YOLO(os.path.join(MODEL_DIR, "yolo11m_sign_language.pt")),
+    "YOLO11m (COCO128)": YOLO("https://huggingface.co/spaces/mbar0075/YOLO-Application-Toolkit/blob/main/models/yolo11m.pt"),
+    "American Sign Language (ASL) YOLO11m": YOLO("https://huggingface.co/spaces/mbar0075/YOLO-Application-Toolkit/blob/main/models/yolo11m_sign_language.pt"),
     # "Microscopic Cell Detection YOLO11m": YOLO(os.path.join(MODEL_DIR, "yolo11m_microscope_cells.pt")),
-    "Website Screenshots YOLO11m": YOLO(os.path.join(MODEL_DIR, "yolo11m_website_screenshots.pt")),
-    "Zoo Animals YOLO11m": YOLO(os.path.join(MODEL_DIR, "yolo11m_zoo_animals.pt")),
-    "Pinned Circuit Boards YOLO11m": YOLO(os.path.join(MODEL_DIR, "yolo11m_circuit_boards.pt")),
+    "Website Screenshots YOLO11m": YOLO("https://huggingface.co/spaces/mbar0075/YOLO-Application-Toolkit/blob/main/models/yolo11m_website_screenshots.pt"),
+    "Zoo Animals YOLO11m": YOLO("https://huggingface.co/spaces/mbar0075/YOLO-Application-Toolkit/blob/main/models/yolo11m_zoo_animals.pt"),
+    "Pinned Circuit Boards YOLO11m": YOLO("https://huggingface.co/spaces/mbar0075/YOLO-Application-Toolkit/blob/main/models/yolo11m_circuit_boards.pt"),
 }
 
 example_dir = "https://huggingface.co/spaces/mbar0075/YOLO-Application-Toolkit/blob/main/"