Update app.py

app.py

@@ -4,10 +4,6 @@ import numpy as np
 from ultralytics import YOLO
 from huggingface_hub import hf_hub_download
 
-model_file = hf_hub_download(repo_id="TheKnight115/Yolov8m", filename="yolov8_Medium.pt")
-
-# Load the YOLO model
-model = YOLO(model_file)
 
 def run_yolo(image):
     # Run the model on the image and get results
@@ -104,6 +100,11 @@ def process_video(uploaded_file):
     st.download_button(label='Download Processed Video', data=video_bytes, file_name='processed_video.mp4', mime='video/mp4')
 
 def main():
+    model_file = hf_hub_download(repo_id="TheKnight115/Yolov8m", filename="yolov8_Medium.pt")
+    
+    # Load the YOLO model
+    model = YOLO(model_file)
+
     st.title("Motorbike Violation Detection")
 
     # Upload file