Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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()
|