enoreyes commited on
Commit
326862d
1 Parent(s): acb7417

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def formatted_message(audio_class, human_input):
98
  llm=OpenAI(temperature=.5, openai_api_key=session_token),
99
  prompt=prompt,
100
  verbose=True,
101
- memory=ConversationalBufferWindowMemory(k=2, ai=audio_class),
102
  )
103
 
104
  output = chatgpt_chain.predict(human_input=message)
 
98
  llm=OpenAI(temperature=.5, openai_api_key=session_token),
99
  prompt=prompt,
100
  verbose=True,
101
+ memory=ConversationalBufferWindowMemory(k=2, ai_prefix=audio_class),
102
  )
103
 
104
  output = chatgpt_chain.predict(human_input=message)