Kirili4ik commited on
Commit
e4ef938
1 Parent(s): ac61535

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -89,7 +89,7 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint)
89
  model = model.eval()
90
 
91
  # Gradio
92
- checkbox_group = gr.CheckboxGroup(['Kirill', 'Me'], default=['Kirill'], type="value", label=None)
93
  title = "Chat with Kirill (in Russian)"
94
  description = "Тут можно поболтать со мной. Но вместо меня бот. Оставь сообщение пустым, чтобы Кирилл продолжил говорить - он очень любит писать подряд несколько сообщений в чате. Используй слайдер, чтобы ответы были более общими или более конкретными (ближе к теме). Подбробнее о технике по ссылке внизу."
95
  article = "<p style='text-align: center'><a href='https://github.com/Kirili4ik/ruDialoGpt3-finetune-colab'>Github with fine-tuning GPT-3 on your chat</a></p>"
@@ -104,9 +104,9 @@ chatbot = gr.Chatbot(color_map=("green", "gray"))
104
  iface = gr.Interface(chat_function,
105
  [
106
  "text",
107
- gr.Radio(["short", "medium", "long"], default='medium'),
108
- gr.Radio(["Kirill", "Me"], default='Kirill'),
109
- gr.Slider(0, 1, default=0.5),
110
  "state"
111
  ],
112
  ["html", "state"],
 
89
  model = model.eval()
90
 
91
  # Gradio
92
+ checkbox_group = gr.inputs.CheckboxGroup(['Kirill', 'Me'], default=['Kirill'], type="value", label=None)
93
  title = "Chat with Kirill (in Russian)"
94
  description = "Тут можно поболтать со мной. Но вместо меня бот. Оставь сообщение пустым, чтобы Кирилл продолжил говорить - он очень любит писать подряд несколько сообщений в чате. Используй слайдер, чтобы ответы были более общими или более конкретными (ближе к теме). Подбробнее о технике по ссылке внизу."
95
  article = "<p style='text-align: center'><a href='https://github.com/Kirili4ik/ruDialoGpt3-finetune-colab'>Github with fine-tuning GPT-3 on your chat</a></p>"
 
104
  iface = gr.Interface(chat_function,
105
  [
106
  "text",
107
+ gr.inputs.Radio(["short", "medium", "long"], default='medium'),
108
+ gr.inputs.Radio(["Kirill", "Me"], default='Kirill'),
109
+ gr.inputs.Slider(0, 1, default=0.5),
110
  "state"
111
  ],
112
  ["html", "state"],