Orami01 commited on
Commit
7b2bb26
·
1 Parent(s): bfcd48d

Update app.py

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