sarasad commited on
Commit
9ef830a
·
1 Parent(s): 6e15d6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -10
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(show_label=False).style(height=650)
44
  msg = gr.Textbox(placeholder="Send a message",show_label=False).style(container=False)
45
- output_slider=gr.Slider(5, 20, step=1, label='Minimum Output Length')
46
- top_p_slider=gr.Slider(0.7, 1, step=0.1, label='Top-P')
47
- temperature_slider=gr.Slider(1, 3, step=0.1, label='Temperature')
48
- def respond(message,chat_history,output_slider,top_p_slider,temperature_slider):
49
- bot_message = generate_response(message,output_slider,top_p_slider,temperature_slider)
50
- chat_history.append((message, bot_message))
51
- return "", chat_history
52
- clear = gr.Button("Clear")
 
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()