mipatov commited on
Commit
67ab8b8
·
1 Parent(s): 5b6f474

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -32,6 +32,7 @@ def predict_gpt(text, model, tokenizer, temperature=1.0):
32
  eos_token_id = tokenizer.eos_token_id,
33
  pad_token_id = tokenizer.pad_token_id,
34
  repetition_penalty = 2.5,
 
35
  num_return_sequences = 1,
36
  output_attentions = True,
37
  return_dict_in_generate=True,
@@ -76,7 +77,7 @@ demo = gr.Interface(
76
  fn=generate,
77
  inputs=[
78
  gr.components.Dropdown(label="Модель", choices=('GPT', 'T5')),
79
- gr.components.Slider(label="Температура",minimum = 0.9,maximum = 3.0,step = 0.1),
80
  gr.components.Textbox(label="Характеристики",value = example),
81
  ],
82
  outputs=[
 
32
  eos_token_id = tokenizer.eos_token_id,
33
  pad_token_id = tokenizer.pad_token_id,
34
  repetition_penalty = 2.5,
35
+ length_penalty = 0.7,
36
  num_return_sequences = 1,
37
  output_attentions = True,
38
  return_dict_in_generate=True,
 
77
  fn=generate,
78
  inputs=[
79
  gr.components.Dropdown(label="Модель", choices=('GPT', 'T5')),
80
+ gr.components.Slider(label="Температура",minimum = 1.0,maximum = 3.0,step = 0.1),
81
  gr.components.Textbox(label="Характеристики",value = example),
82
  ],
83
  outputs=[