dennyaw commited on
Commit
f3adec8
1 Parent(s): a7a03e7

updated gradio interface

Browse files
Files changed (1) hide show
  1. app.py +20 -26
app.py CHANGED
@@ -92,18 +92,10 @@ def bot(history,temperature, max_new_tokens, top_p,top_k):
92
  do_sample=True
93
  )
94
 
95
- # convert the tokens to text, and then split the responses into lines
96
  response = tokenizer.batch_decode(response, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
97
- #response = response.split("### Tanggapan:")
98
- #response = response[1].strip() if len(response) > 1 else ""
99
  sections = response.split("###")
100
-
101
- # Ambil potongan yang berisi "Tanggapan" yang pertama
102
  response = sections[3]
103
-
104
- #return response.split("Tanggapan:")[1].strip()
105
- #history[-1] = response.split("Tanggapan:")[1].strip()
106
- #return history
107
  response=response.split("Tanggapan:")[1].strip()
108
  history[-1][1] = response
109
  return history
@@ -114,30 +106,32 @@ with gr.Blocks() as demo:
114
  """# ChatDoctor - PolyLM 1.7b 🩺
115
 
116
  A [ChatDoctor - PolyLM 1.7b](https://huggingface.co/fadliaulawi/polylm-1.7b-finetuned) demo.
117
- From the [PolyLM 1.7b](https://huggingface.co/DAMO-NLP-MT/polylm-1.7b) model and finetuned on [ChatDoctor](https://github.com/Kent0n-Li/ChatDoctor) dataset.
118
 
119
  """
120
  )
121
- with gr.Row() as row:
122
- with gr.Column():
123
- msg = gr.Textbox()
124
- temperature = gr.Slider(0, 5, value=0.8, step=0.1, label='Temperature')
125
- max_length = gr.Slider(0, 1024, value=50, step=1, label='Max Length')
126
- top_p = gr.Slider(0, 1, value=0.8, step=0.1, label='Top P')
127
- top_k = gr.Slider(0, 50, value=10, step=1, label='Top K')
128
- submit = gr.Button("Submit")
129
- clear = gr.Button("Clear")
130
- with gr.Column():
131
- chatbot = gr.Chatbot()
132
- examples = gr.Examples(examples=["Dokter, aku merasa sangat depresi akhir-akhir ini dan juga mengalami perubahan suhu tubuhku.",
133
- "Dokter, aku mengalami kelelahan akhir-akhir ini.",
134
- "Dokter, aku merasa pusing, lemah dan sakit dada tajam akhir-akhir ini.",
135
  "Dokter, saya sudah beberapa minggu mengalami suara serak dan tidak kunjung membaik meski sudah minum obat. Apa masalahnya?"
136
  ],inputs=[msg])
 
 
 
 
 
 
 
 
 
137
 
138
- submit.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
139
  bot, [chatbot,temperature,max_length,top_p,top_k], chatbot
140
  )
141
- clear.click(lambda: None, None, chatbot, queue=False)
142
 
143
  demo.queue(concurrency_count=100).launch()
 
92
  do_sample=True
93
  )
94
 
95
+ # clean up response before returning
96
  response = tokenizer.batch_decode(response, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
 
 
97
  sections = response.split("###")
 
 
98
  response = sections[3]
 
 
 
 
99
  response=response.split("Tanggapan:")[1].strip()
100
  history[-1][1] = response
101
  return history
 
106
  """# ChatDoctor - PolyLM 1.7b 🩺
107
 
108
  A [ChatDoctor - PolyLM 1.7b](https://huggingface.co/fadliaulawi/polylm-1.7b-finetuned) demo.
109
+ From the [PolyLM 1.7b](https://huggingface.co/DAMO-NLP-MT/polylm-1.7b) model and finetuned on the Indonesian translation of [ChatDoctor](https://github.com/Kent0n-Li/ChatDoctor) dataset.
110
 
111
  """
112
  )
113
+
114
+ chatbot = gr.Chatbot()
115
+ msg = gr.Textbox()
116
+ submit = gr.Button("Submit")
117
+ clear = gr.Button("Clear")
118
+ examples = gr.Examples(examples=["Dokter, aku mengalami kelelahan akhir-akhir ini.", "Dokter, aku merasa pusing, lemah dan sakit dada tajam akhir-akhir ini.",
119
+ "Dokter, aku merasa sangat depresi akhir-akhir ini dan juga mengalami perubahan suhu tubuhku.",
 
 
 
 
 
 
 
120
  "Dokter, saya sudah beberapa minggu mengalami suara serak dan tidak kunjung membaik meski sudah minum obat. Apa masalahnya?"
121
  ],inputs=[msg])
122
+
123
+ gr.Markdown(
124
+ """## Adjust the additional inputs:"""
125
+ )
126
+
127
+ temperature = gr.Slider(0, 5, value=0.8, step=0.1, label='Temperature',info="Controls randomness, higher values increase diversity.")
128
+ max_length = gr.Slider(0, 1024, value=50, step=1, label='Max Length',info="The maximum numbers of output's tokens.")
129
+ top_p = gr.Slider(0, 1, value=0.8, step=0.1, label='Top P',info="The cumulative probability cutoff for token selection. Lower values mean sampling from a smaller, more top-weighted nucleus.")
130
+ top_k = gr.Slider(0, 50, value=10, step=1, label='Top K',info="Sample from the k most likely next tokens at each step. Lower k focuses on higher probability tokens.")
131
 
132
+ submit.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
133
  bot, [chatbot,temperature,max_length,top_p,top_k], chatbot
134
  )
135
+ clear.click(lambda: None, None, chatbot, queue=False)
136
 
137
  demo.queue(concurrency_count=100).launch()