Update app.py

app.py

@@ -5,8 +5,8 @@ from transformers import AutoTokenizer, AutoModel
 # Load the tokenizer and model
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained('dwzhu/e5-base-4k', trust_remote_code=True)
-    model = AutoModel.from_pretrained('dwzhu/e5-base-4k', trust_remote_code=True)
+    tokenizer = AutoTokenizer.from_pretrained('Alibaba-NLP/gte-base-en-v1.5', trust_remote_code=True)
+    model = AutoModel.from_pretrained('Alibaba-NLP/gte-base-en-v1.5', trust_remote_code=True)
     model.to('cpu')
     return tokenizer, model