Orami01 commited on
Commit
5531ee9
·
1 Parent(s): e4c99ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -36,10 +36,13 @@ docsearch.save_local(DB_FAISS_PATH)
36
 
37
  from transformers import pipeline
38
 
39
- pipe = pipeline("text-generation", model="meta-llama/Llama-2-7b-chat-hf")
40
- model_type="llama",
41
- max_new_tokens=512,
42
- temperature=0.1)
 
 
 
43
 
44
  qa = ConversationalRetrievalChain.from_llm(llm, retriever=docsearch.as_retriever())
45
 
 
36
 
37
  from transformers import pipeline
38
 
39
+ pipe = pipeline(
40
+ "text-generation",
41
+ model="meta-llama/Llama-2-7b-chat-hf",
42
+ model_type="llama",
43
+ max_new_tokens=512,
44
+ temperature=0.1
45
+ )
46
 
47
  qa = ConversationalRetrievalChain.from_llm(llm, retriever=docsearch.as_retriever())
48