samim2024 commited on
Commit
9ad6ba3
·
verified ·
1 Parent(s): 34f198a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -20,7 +20,10 @@ from langchain_community.llms import HuggingFaceEndpoint
20
  #The PaLM 2 for Text (text-bison, text-unicorn) foundation models are optimized for a variety of natural language
21
  #tasks such as sentiment analysis, entity extraction, and content creation. The types of content that the PaLM 2 for
22
  #Text models can create include document summaries, answers to questions, and labels that classify content.
 
23
  llm = HuggingFaceEndpoint(repo_id="mistralai/Mistral-7B-Instruct-v0.2", Temperature=0.9)
 
 
24
  #llm = VertexAI(model_name="text-bison@001",max_output_tokens=256,temperature=0.1,top_p=0.8,top_k=40,verbose=True,)
25
 
26
  #embeddings = VertexAIEmbeddings()
 
20
  #The PaLM 2 for Text (text-bison, text-unicorn) foundation models are optimized for a variety of natural language
21
  #tasks such as sentiment analysis, entity extraction, and content creation. The types of content that the PaLM 2 for
22
  #Text models can create include document summaries, answers to questions, and labels that classify content.
23
+
24
  llm = HuggingFaceEndpoint(repo_id="mistralai/Mistral-7B-Instruct-v0.2", Temperature=0.9)
25
+ model = SentenceTransformer("all-MiniLM-L6-v2")
26
+
27
  #llm = VertexAI(model_name="text-bison@001",max_output_tokens=256,temperature=0.1,top_p=0.8,top_k=40,verbose=True,)
28
 
29
  #embeddings = VertexAIEmbeddings()