peterkros commited on
Commit
b1746f2
·
verified ·
1 Parent(s): 14645ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -49,12 +49,17 @@ def generate_response(message, selected_topics):
49
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
50
  return response
51
 
52
- # Define Gradio interface
53
- iface = gr.Interface(fn=generate_response,
54
- inputs=[gr.components.Chatbot(), gr.inputs.Dropdown(choices=topics_list, label="Select Topics", allow_multiple=True)],
55
- outputs="text",
56
- title="Child-Safe Chatbot BETA",
57
- description="A chatbot that stays on topic and filters inappropriate content. Select relevant topics.")
 
 
 
 
 
58
 
59
  # Run the app
60
  if __name__ == "__main__":
 
49
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
50
  return response
51
 
52
+
53
+
54
+
55
+ dropdown = gr.components.Dropdown(choices=topics_list, label="Select Topics", allow_multiple=True)
56
+
57
+ iface = gr.ChatInterface(
58
+ fn=generate_response,
59
+ title="Child-Safe Chatbot BETA",
60
+ additional_inputs=[dropdown],
61
+ description="A chatbot that stays on topic and filters inappropriate content. Select relevant topics.",
62
+ )
63
 
64
  # Run the app
65
  if __name__ == "__main__":