warleagle commited on
Commit
a3facf2
·
verified ·
1 Parent(s): f66cce3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -45,16 +45,23 @@ def predict(input_data, temp):
45
  conversation.add_user_message(input_data)
46
  prompt = conversation.get_prompt()
47
 
48
- output = generate(model, tokenizer, prompt, generation_config)
49
- return output
 
 
 
50
 
51
  io = gr.Interface(predict,
52
- inputs=[gr.Text(value="Как записаться к стоматологу?"),
 
53
  gr.Slider(minimum=0.01,
54
  maximum=1,
55
  value=0.3,
56
- step=0.1)],
57
- outputs=[gr.Text()])
 
 
 
58
 
59
  if __name__ == "__main__":
60
  io.launch()
 
45
  conversation.add_user_message(input_data)
46
  prompt = conversation.get_prompt()
47
 
48
+ output_one = generate(model, tokenizer, prompt, generation_config)
49
+ output_two = generate(model, tokenizer, prompt, generation_config)
50
+ output_three = generate(model, tokenizer, prompt, generation_config)
51
+
52
+ return output_one, output_two, output_three
53
 
54
  io = gr.Interface(predict,
55
+ inputs=[gr.Textbox(value="Как записаться к стоматологу?",
56
+ label="Введите текст:"),
57
  gr.Slider(minimum=0.01,
58
  maximum=1,
59
  value=0.3,
60
+ step=0.1,
61
+ info="Данный параметр позволяет изменять креативность модели. Чем больше, тем модель будет более креативная и наоборот.")],
62
+ outputs=[gr.Textbox(label="Первый вариант ответа:"),
63
+ gr.Textbox(label="Второй вариант ответа:"),
64
+ gr.Textbox(label="Третий вариант ответа:")])
65
 
66
  if __name__ == "__main__":
67
  io.launch()