mathslearn commited on
Commit
d1385b6
·
verified ·
1 Parent(s): 9198c4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 st.session_state:
19
- st.session_state.buffer_memory = ConversationBufferWindowMemory(k=8)
20
- #st.session_state.buffer_memory = ConversationBufferWindowMemory(k=3)
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,