debug

app.py

@@ -138,7 +138,7 @@ class VideoHighlightDetector:
         
         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]
-        
+        print(f"Segment response {response}")
         return "yes" in response
 
     def _concatenate_scenes(
@@ -350,6 +350,7 @@ def create_ui(examples_path: str, model_path: str):
 
                     # Process segment
                     if detector.process_segment(segment_path, highlights):
+                        print("KEEPING SEGMENT")
                         kept_segments.append((start_time, end_time))
                     
                     # Clean up segment file