fix

app.py

@@ -81,7 +81,7 @@ class VideoHighlightDetector:
         ).to(self.device)
         
         outputs = self.model.generate(**inputs, max_new_tokens=512, do_sample=True, temperature=0.7)
-        return self.processor.decode(outputs[0], skip_special_tokens=True).lower().split("Assistant: ")[1]
+        return self.processor.decode(outputs[0], skip_special_tokens=True).lower().split("assistant: ")[1]
 
     def determine_highlights(self, video_description: str) -> str:
         """Determine what constitutes highlights based on video description."""
@@ -109,7 +109,7 @@ class VideoHighlightDetector:
         ).to(self.device)
         
         outputs = self.model.generate(**inputs, max_new_tokens=256, do_sample=True, temperature=0.7)
-        return self.processor.decode(outputs[0], skip_special_tokens=True).lower().split("Assistant: ")[1]
+        return self.processor.decode(outputs[0], skip_special_tokens=True).split("Assistant: ")[1]
 
     def process_segment(self, video_path: str, highlight_types: str) -> bool:
         """Process a video segment and determine if it contains highlights."""
@@ -137,7 +137,7 @@ class VideoHighlightDetector:
         ).to(self.device)
         
         outputs = self.model.generate(**inputs, max_new_tokens=64, do_sample=False)
-        response = self.processor.decode(outputs[0], skip_special_tokens=True).lower().split("Assistant: ")[1]
+        response = self.processor.decode(outputs[0], skip_special_tokens=True).lower().split("assistant: ")[1]
         
         return "yes" in response