Update app.py

app.py

@@ -3,9 +3,20 @@ from ultralytics import YOLO
 import cv2
 import os
 import uuid
+import os
+import torch
+from ultralytics import YOLO
+
+# Automatically download your best.pt model from your dataset repo
+model_path = "best.pt"
+if not os.path.exists(model_path):
+    os.system("wget https://huggingface.co/datasets/Prasanna1622/solar-fault-dataset/resolve/main/best.pt")
+
+# Load the model
+model = YOLO(model_path)
+
+# Your Gradio/Streamlit app here...
 
-# Load YOLOv8 model (make sure the file is uploaded)
-model = YOLO("best.pt")  # Replace with your actual model name if different
 
 def detect_faults(video):
     temp_dir = "temp_" + str(uuid.uuid4())