minor bux fixed

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import json
 from src.pipeline import run_pipeline, search_highlights
+from dotenv import load_dotenv
+
+load_dotenv() 
 
 def extract(video_file, game_card_str):
     video_file.save("uploaded.mp4")

src/embedder.py

@@ -6,7 +6,7 @@ import decord
 class InternVLEmbedder:
     def __init__(self):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.model = AutoModel.from_pretrained("OpenGVLab/InternVL2_5-8B-MPO").to(self.device)
+        self.model = AutoModel.from_pretrained("OpenGVLab/InternVL2_5-8B-MPO", trust_remote_code=True).to(self.device)
         self.processor = AutoProcessor.from_pretrained("OpenGVLab/InternVL2_5-8B-MPO")
 
     def embed_video(self, video_path):

src/labeler.py

@@ -1,5 +1,8 @@
 import os
 from together import Together
+from dotenv import load_dotenv
+
+load_dotenv() 
 
 class TogetherLLMLabeler:
     def __init__(self):

src/pinecone_store.py

@@ -1,5 +1,8 @@
 import pinecone
 import os
+from dotenv import load_dotenv
+
+load_dotenv() 
 
 class PineconeStore:
     def __init__(self):

src/segmenter.py

@@ -1,7 +1,9 @@
 import cv2
 import os
 from roboflow import Roboflow
+from dotenv import load_dotenv
 
+load_dotenv() 
 ## When the ball is no longer detected, we start a new segment
 
 def detect_event_segments(video_path, confidence=0.4):

src/utils.py

@@ -1,5 +1,8 @@
 import cv2
 import os
+from dotenv import load_dotenv
+
+load_dotenv() 
 
 def extract_key_frames(video_path, start_sec, end_sec):
     cap = cv2.VideoCapture(video_path)