Spaces:
Runtime error
Runtime error
Commit
·
c729c7f
1
Parent(s):
71cc93d
Changed the text generation model
Browse files
app.py
CHANGED
@@ -69,7 +69,7 @@ def get_vectorstore(text_chunks, index):
|
|
69 |
|
70 |
|
71 |
def get_conversation_chain(vectorstore):
|
72 |
-
llm = HuggingFaceHub(repo_id="
|
73 |
|
74 |
memory = ConversationBufferMemory(
|
75 |
memory_key='chat_history', return_messages=True)
|
|
|
69 |
|
70 |
|
71 |
def get_conversation_chain(vectorstore):
|
72 |
+
llm = HuggingFaceHub(repo_id="mistralai/Mistral-7B-v0.1", model_kwargs={"temperature":0.2, "max_length":1024})
|
73 |
|
74 |
memory = ConversationBufferMemory(
|
75 |
memory_key='chat_history', return_messages=True)
|