Update main.py

main.py

@@ -9,7 +9,7 @@ from ultralytics import YOLO
 from openpyxl import Workbook
 
 # Ensure necessary directories exist
-save_path = "/home/ml/ML/ml_backup/arjun/"
+save_path = "./"
 screenshots_path = os.path.join(save_path, "screenshots")
 detect_path = os.path.join(save_path, "runs/detect/")
 
@@ -20,7 +20,7 @@ os.makedirs(screenshots_path, exist_ok=True)
 classes = ['Head and shoulders bottom', 'Head and shoulders top', 'M_Head', 'StockLine', 'Triangle', 'W_Bottom']
 
 # Load YOLOv8 model
-model_path = "/home/ml/ML/ml_backup/arjun/best111.pt"
+model_path = "model.pt"
 if not os.path.exists(model_path):
     raise FileNotFoundError(f"Model file not found: {model_path}")
 model = YOLO(model_path)
@@ -35,7 +35,7 @@ ws = wb.active
 ws.append(["Timestamp", "Predicted Image Path", "Label"])  # Headers
 
 # Initialize video writer
-video_path = os.path.join(save_path, "annotated_video.mp4")
+video_path = "./videos/annotated_video.mp4"
 fourcc = cv2.VideoWriter_fourcc(*"mp4v")
 fps = 0.5  # Adjust frames per second as needed
 video_writer = None