Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,9 +15,9 @@ tokenizer = AutoTokenizer.from_pretrained(REPO_ID)
|
|
| 15 |
|
| 16 |
llm = ChatOpenAI(model=model, tokenizer=tokenizer)
|
| 17 |
|
| 18 |
-
if 'buffer_memory' not in
|
| 19 |
-
|
| 20 |
-
#
|
| 21 |
|
| 22 |
conversation = ConversationChain(
|
| 23 |
llm=llm,
|
|
|
|
| 15 |
|
| 16 |
llm = ChatOpenAI(model=model, tokenizer=tokenizer)
|
| 17 |
|
| 18 |
+
if 'buffer_memory' not in gr.session_state:
|
| 19 |
+
gr.session_state.buffer_memory = ConversationBufferWindowMemory(k=8)
|
| 20 |
+
#gr.session_state.buffer_memory = ConversationBufferWindowMemory(k=3)
|
| 21 |
|
| 22 |
conversation = ConversationChain(
|
| 23 |
llm=llm,
|