Update main.py

rishiraj · Aug 15, 2024 · 58c094e · 58c094e
1 parent 515da19
commit 58c094e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spanking/main.py b/spanking/main.py
@@ -8,7 +8,7 @@
 import pandas as pd
 
 class VectorDB:
-    def __init__(self, model_name='Alibaba-NLP/gte-base-en-v1.5'):
+    def __init__(self, model_name='avsolatorio/GIST-small-Embedding-v0'):
         self.model = SentenceTransformer(model_name, trust_remote_code=True)
         self.image_classifier = pipeline(task="zero-shot-image-classification", model="google/siglip-so400m-patch14-384")
         self.texts = []