mathslearn commited on
Commit
507cfea
·
verified ·
1 Parent(s): ead1bee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -17,13 +17,13 @@ import gradio as gr
17
 
18
  llm = ChatOpenAI(model_name="gpt-3.5-turbo", openai_api_key="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")
19
 
20
- if 'buffer_memory' not in gr.session_state:
21
- gr.session_state.buffer_memory = ConversationBufferWindowMemory(k=8)
22
- #gr.session_state.buffer_memory = ConversationBufferWindowMemory(k=3)
23
 
24
  conversation = ConversationChain(
25
  llm=llm,
26
- memory=gr.session_state.buffer_memory,
27
  verbose=True
28
  )
29
 
 
17
 
18
  llm = ChatOpenAI(model_name="gpt-3.5-turbo", openai_api_key="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")
19
 
20
+ # Initialize global variable for conversation memory
21
+ buffer_memory = ConversationBufferWindowMemory(k=8)
22
+ #buffer_memory = ConversationBufferWindowMemory(k=3)
23
 
24
  conversation = ConversationChain(
25
  llm=llm,
26
+ memory=buffer_memory,
27
  verbose=True
28
  )
29