Update app.py

app.py

@@ -12,16 +12,17 @@ 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")
 
+# Initialize the YOLO model
+model = YOLO(model_path)
 
-
-        # Inference function
+# Inference function
 def detect_faults(video_path):
     """
     - video_path: the path to the uploaded video file on disk.
     - Returns: path to the annotated output.mp4.
     """
     try:
-        # Create a unique RUN directory so YOLO does overwrite previous results
+        # Create a unique RUN directory so YOLO does not overwrite previous results
         unique_id = str(uuid.uuid4())[:8]
         project_dir = os.path.join("runs", "detect", unique_id)
         os.makedirs(project_dir, exist_ok=True)
@@ -29,7 +30,7 @@ def detect_faults(video_path):
 
         # Run YOLO predict; this saves the annotated video in project_dir/
         results = model.predict(
-            source=video_path,    # path to the uploaded video
+            source=video_path,  # path to the uploaded video
             save=True,
             save_txt=False,
             conf=0.5,
@@ -38,8 +39,7 @@ def detect_faults(video_path):
         )
         print("✅ YOLO predict() finished.")
 
-        # YOLO’s default behavior: 
-        #   annotated video is saved in runs/detect/<unique_id>/<original_filename>
+        # Check if output video exists
         original_name = os.path.basename(video_path)
         output_video_path = os.path.join("runs", "detect", unique_id, original_name)
         print(f"🛠️  Looking for output video at: {output_video_path}")