Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import transformers
|
|
2 |
import gradio as gr
|
3 |
import git
|
4 |
import os
|
|
|
5 |
os.system("pip install --upgrade pip")
|
6 |
|
7 |
#Load arabert preprocessor
|
@@ -38,18 +39,20 @@ def generate_response(text, minimum_length, p, temperature):
|
|
38 |
# description = 'This demo is for a BERT2BERT model trained for single-turn open-domain empathetic dialogue response generation in Modern Standard Arabic'
|
39 |
|
40 |
with gr.Blocks() as demo:
|
41 |
-
|
42 |
gr.Markdown("Empathetic Response Generation in Arabic")
|
43 |
-
chatbot = gr.Chatbot(
|
44 |
msg = gr.Textbox(placeholder="Send a message",show_label=False).style(container=False)
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
|
|
53 |
msg.submit(respond, [msg, chatbot,output_slider,top_p_slider,temperature_slider], [msg, chatbot])
|
54 |
clear.click(lambda: None, None, chatbot, queue=False)
|
|
|
55 |
demo.launch()
|
|
|
2 |
import gradio as gr
|
3 |
import git
|
4 |
import os
|
5 |
+
from gradio import rtl
|
6 |
os.system("pip install --upgrade pip")
|
7 |
|
8 |
#Load arabert preprocessor
|
|
|
39 |
# description = 'This demo is for a BERT2BERT model trained for single-turn open-domain empathetic dialogue response generation in Modern Standard Arabic'
|
40 |
|
41 |
with gr.Blocks() as demo:
|
42 |
+
|
43 |
gr.Markdown("Empathetic Response Generation in Arabic")
|
44 |
+
chatbot = gr.Chatbot().style(height=400)
|
45 |
msg = gr.Textbox(placeholder="Send a message",show_label=False).style(container=False)
|
46 |
+
with gr.Column():
|
47 |
+
output_slider=gr.Slider(5, 20, step=1, label='Minimum Output Length')
|
48 |
+
top_p_slider=gr.Slider(0.7, 1, step=0.1, label='Top-P')
|
49 |
+
temperature_slider=gr.Slider(1, 3, step=0.1, label='Temperature')
|
50 |
+
def respond(message,chat_history,output_slider,top_p_slider,temperature_slider):
|
51 |
+
bot_message = generate_response(message,output_slider,top_p_slider,temperature_slider)
|
52 |
+
chat_history.append((message, bot_message))
|
53 |
+
return "", chat_history
|
54 |
+
clear = gr.Button("Clear Chat")
|
55 |
msg.submit(respond, [msg, chatbot,output_slider,top_p_slider,temperature_slider], [msg, chatbot])
|
56 |
clear.click(lambda: None, None, chatbot, queue=False)
|
57 |
+
demo.rtl=True
|
58 |
demo.launch()
|