medmac01 commited on
Commit
dd95821
1 Parent(s): c60bc54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def get_vectorstore(text_chunks):
39
 
40
  def get_conversation_chain(vectorstore):
41
  # llm = ChatOpenAI()
42
- llm = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.5, "max_length":512})
43
 
44
  memory = ConversationBufferMemory(
45
  memory_key='chat_history', return_messages=True)
 
39
 
40
  def get_conversation_chain(vectorstore):
41
  # llm = ChatOpenAI()
42
+ llm = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.5, "max_length":512, "return_full_text":False})
43
 
44
  memory = ConversationBufferMemory(
45
  memory_key='chat_history', return_messages=True)