ashwinR commited on
Commit
fe57737
1 Parent(s): bbbec14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ def generate_response(question):
21
  #response = llm_chain.run(question)
22
  question = [{"role": "user", "content" : question}]
23
  response = gptj.chat_completion(question)
 
24
  response = response if response else "Sorry, Try Again"
25
  return response
26
 
 
21
  #response = llm_chain.run(question)
22
  question = [{"role": "user", "content" : question}]
23
  response = gptj.chat_completion(question)
24
+ response = response['choices'][0]['message']['content']
25
  response = response if response else "Sorry, Try Again"
26
  return response
27