Update app.py

app.py

@@ -14,19 +14,25 @@ import os
 # from transformers import AutoModelForCausalLM, AutoTokenizer
 # from transformers import pipeline
 
+from transformers import AutoModel
+llm = AutoModel.from_pretrained("Alibaba-NLP/gte-large-en-v1.5", trust_remote_code=True) 
+
+# LLAMA
 # hf_token = os.environ['llama_token']
 
 # login(token=hf_token)
 
 # llm = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B-Instruct")
-
 # llm = pipeline("text-generation", model = "meta-llama/Meta-Llama-3-70B-Instruct")
 
+# MISTRAL
 # llm = pipeline("text-generation", model="mistralai/Mixtral-8x22B-Instruct-v0.1")
 
-api_key = "AIzaSyCdM_aAIsW_nPbjarOF83mbX1_z1cVX2_M"
 
-llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key)
+# TO USE GOOGLE MODELS
+# api_key = "AIzaSyCdM_aAIsW_nPbjarOF83mbX1_z1cVX2_M"
+
+# llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key)
 # llm = GooglePalm(google_api_key = api_key, temperature=0.7)
 
 # LOADING CSV FILE