d0r1h commited on
Commit
ac78da2
·
1 Parent(s): a828481

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def predict(input, history=[]):
11
  history = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id).tolist()
12
  response = tokenizer.decode(history[0]).replace("<|endoftext|>", "\n")
13
 
14
- return response, history
15
 
16
  gr.Interface(fn=predict, title="DialoGPT-large", inputs=[gr.inputs.Textbox(placeholder="Write a text"), "state"],
17
  outputs=[gr.outputs.Textbox(label="Output"), "state"]).launch()
 
11
  history = model.generate(bot_input_ids, max_length=500, pad_token_id=tokenizer.eos_token_id).tolist()
12
  response = tokenizer.decode(history[0]).replace("<|endoftext|>", "\n")
13
 
14
+ return response
15
 
16
  gr.Interface(fn=predict, title="DialoGPT-large", inputs=[gr.inputs.Textbox(placeholder="Write a text"), "state"],
17
  outputs=[gr.outputs.Textbox(label="Output"), "state"]).launch()