Update app.py

app.py

@@ -11,13 +11,16 @@ from huggingface_hub import login
 import os
 
 
-# from transformers import AutoModelForCausalLM, AutoTokenizer
-# from transformers import pipeline
+from transformers import pipeline
+pipe = pipeline("feature-extraction", model="mixedbread-ai/mxbai-embed-large-v1") 
 
-from transformers import AutoModel
-llm = AutoModel.from_pretrained("Alibaba-NLP/gte-large-en-v1.5", trust_remote_code=True) 
+# from transformers import AutoModel
+# llm = AutoModel.from_pretrained("Alibaba-NLP/gte-large-en-v1.5", trust_remote_code=True) 
 
 # LLAMA
+# from transformers import AutoModelForCausalLM, AutoTokenizer
+# from transformers import pipeline
+
 # hf_token = os.environ['llama_token']
 
 # login(token=hf_token)