RoAr777 commited on
Commit
18e3a2b
·
1 Parent(s): b1bd66f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,5 +15,5 @@ def chatbot_model(prompt):
15
  decoded_output = tokenizer.decode(output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
16
  return decoded_output
17
 
18
- iface = gr.Interface(fn=chatbot_model, inputs="text", outputs="text")
19
  iface.launch()
 
15
  decoded_output = tokenizer.decode(output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
16
  return decoded_output
17
 
18
+ iface = gr.ChatInterface(chatbot_model)
19
  iface.launch()