return more frames

video_semantic_search/app.py

@@ -10,6 +10,8 @@ import streamlit as st
 
 from pipeline import clip_wrapper
 
+NUM_FRAMES_TO_RETURN = 21
+
 
 class SemanticSearcher:
     def __init__(self, dataset: pd.DataFrame):
@@ -22,7 +24,7 @@ class SemanticSearcher:
 
     def search(self, query: str) -> list["SearchResult"]:
         v = self.embedder([query]).detach().numpy()
-        D, I = self.index.search(v, 10)
+        D, I = self.index.search(v, NUM_FRAMES_TO_RETURN)
         return [
             SearchResult(
                 video_id=row["video_id"],