Update app.py

app.py

@@ -15,7 +15,7 @@ os.environ["GOOGLE_API_KEY"] = "AIzaSyDT0y1kJqgGKiOYiYFMXc-2kTgV_WLbOpA"#os.gete
 gemini_model = ChatGoogleGenerativeAI(model="gemini-1.5-flash")
 
 # ✅ Load the YOLO model
-yolo_model = YOLO("/content/Bottle_lable_checking_using_gemini_yolo/best.pt")  
+yolo_model = YOLO("best.pt")  
 names = yolo_model.names  
 
 def encode_image_to_base64(image):
@@ -121,7 +121,7 @@ def gradio_interface(video_path):
     return process_video(video_path)
 
 # ✅ Sample video file
-sample_video_path = "/content/Bottle_lable_checking_using_gemini_yolo/vid4.mp4"  # Make sure this file is available in the working directory
+sample_video_path = "vid4.mp4"  # Make sure this file is available in the working directory
 
 # ✅ Gradio UI setup with sample video
 iface = gr.Interface(