Update app.py

app.py

@@ -1,33 +1,10 @@
 import gradio as gr
-import torch
-import os
-
 from ultralytics import YOLO
-model = YOLO("best.pt")
-
-# --- Configuration ---
-MODEL_REPO_ID = "Nawinkumar15/my-yolov8-model"  # Replace with YOUR model repository ID
-MODEL_FILENAME = "best.pt"
-CACHE_DIR = "model_cache"
 
-# Ensure cache directory exists
-os.makedirs(CACHE_DIR, exist_ok=True)
-
-# Function to load the YOLOv8 model from Hugging Face Hub
-def load_model():
-    try:
-        model = torch.hub.load('ultralytics/yolov8', 'custom', path=f'{CACHE_DIR}/{MODEL_FILENAME}', trust_repo=True)
-        return model
-    except Exception as e:
-        print(f"Error loading model: {e}")
-        return None
-
-# Load the model when the app starts
-model = load_model()
+# ✅ Load model directly from local file (same folder as app.py)
+model = YOLO("best.pt")
 
 def predict(image):
-    if model is None:
-        return "Model not loaded."
     results = model(image)
     return results[0].plot()
 
@@ -39,4 +16,4 @@ iface = gr.Interface(
     description="Upload an image to detect objects using your YOLOv8 model trained with Roboflow."
 )
 
-iface.launch()
+iface.launch()