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)
|