update requirements

rishiraj · Aug 15, 2024 · 49f4f5d · 49f4f5d
1 parent bb4652e
commit 49f4f5d
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 9 deletions.
diff --git a/README.md b/README.md
@@ -2,8 +2,16 @@
 
 <table>
   <tr>
-    <td><img src="/assets/home.png"/></td>
-    <td><img src="/assets/search.png"/></td>
+    <td>
+      <a href="http://127.0.0.1:5000">
+        <img src="/assets/home.png"/>
+      </a>
+    </td>
+    <td>
+      <a href="http://127.0.0.1:5000/search">
+        <img src="/assets/search.png"/>
+      </a>
+    </td>
   </tr>
 </table>
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,9 @@
-Flask
+Flask==3.0.3
 jax==0.4.31
 jaxlib==0.4.31
-sentence-transformers==2.7.0
+sentence-transformers==3.0.1
 sentencepiece==0.2.0
-pandas==2.0.3
-transformers==4.40.1
-Pillow==9.4.0
-requests==2.31.0
+pandas==2.2.2
+transformers==4.44.0
+Pillow==10.4.0
+requests==2.32.3
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='dunzhang/stella_en_400M_v5'):
+    def __init__(self, model_name='Alibaba-NLP/gte-base-en-v1.5'):
         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 = []