v1

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 model = SentenceTransformer('rufimelo/Legal-SBERTimbau-sts-base')
 videos_db = pd.read_csv('videos_db.tsv', header=None, names=["id", "title", "thumb"], sep='\t')
 video_index = hnswlib.Index('cosine', dim=model.get_sentence_embedding_dimension())
-video_index.load_index('video_index.bin')
+video_index.load_index('index.bin')
 
 def predict(query):
     query_embedding = model.encode([query])