DHEIVER commited on
Commit
a004f60
1 Parent(s): f252896

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ def chatbot(question):
12
  response = query({
13
  "inputs": question
14
  })
15
- return response["generated_text"].strip()
 
16
 
17
  iface = gr.Interface(
18
  fn=chatbot,
 
12
  response = query({
13
  "inputs": question
14
  })
15
+ generated_text = response[0]["generated_text"].strip()
16
+ return generated_text
17
 
18
  iface = gr.Interface(
19
  fn=chatbot,