Geraldine commited on
Commit
cb6043a
·
1 Parent(s): adf7988

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,9 +23,9 @@ def define_embeddings_llm(openai_key):
23
  embeddings = HuggingFaceEmbeddings(
24
  model_name="sentence-transformers/all-MiniLM-L6-v2"
25
  )
26
- model_id = "lmsys/vicuna-7b-v1.3"
27
- pipe = pipeline("text-generation", model=model_id, max_length=2048)
28
- llm = HuggingFacePipeline(pipeline=pipe)
29
  langchain.llm_cache = InMemoryCache()
30
  return embeddings,llm
31
 
 
23
  embeddings = HuggingFaceEmbeddings(
24
  model_name="sentence-transformers/all-MiniLM-L6-v2"
25
  )
26
+ HuggingFaceHub(repo_id="MBZUAI/LaMini-Flan-T5-248M" ,
27
+ model_kwargs={"max_length":2048,"do_sample":True,
28
+ "temperature":0.2})
29
  langchain.llm_cache = InMemoryCache()
30
  return embeddings,llm
31