mipatov commited on
Commit
35a6465
·
1 Parent(s): b42b839

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,7 +28,7 @@ def predict_gpt(text, model, tokenizer, temperature=1.0):
28
  num_beams=4,
29
  temperature= temperature,
30
  top_p=0.75,
31
- max_length=356,
32
  length_penalty = 1.5,
33
  eos_token_id = tokenizer.eos_token_id,
34
  pad_token_id = tokenizer.pad_token_id,
@@ -48,7 +48,7 @@ def predict_t5(text, model, tokenizer, temperature=1.2):
48
  num_beams=4,
49
  temperature=temperature,
50
  top_p=0.35,
51
- max_length=356,
52
  length_penalty = -1.0,
53
  output_attentions = True,
54
  return_dict_in_generate=True,
@@ -76,7 +76,7 @@ demo = gr.Interface(
76
  fn=generate,
77
  inputs=[
78
  gr.components.Dropdown(label="Модель", choices=('GPT', 'T5')),
79
- gr.components.Slider(label="Вариативность",minimum = 1.0,maximum = 3.0,step = 0.1),
80
  gr.components.Textbox(label="Характеристики",value = example),
81
  ],
82
  outputs=[
 
28
  num_beams=4,
29
  temperature= temperature,
30
  top_p=0.75,
31
+ max_length=512,
32
  length_penalty = 1.5,
33
  eos_token_id = tokenizer.eos_token_id,
34
  pad_token_id = tokenizer.pad_token_id,
 
48
  num_beams=4,
49
  temperature=temperature,
50
  top_p=0.35,
51
+ max_length=512,
52
  length_penalty = -1.0,
53
  output_attentions = True,
54
  return_dict_in_generate=True,
 
76
  fn=generate,
77
  inputs=[
78
  gr.components.Dropdown(label="Модель", choices=('GPT', 'T5')),
79
+ gr.components.Slider(label="Вариативность",value = 1.2,minimum = 1.0,maximum = 3.0,step = 0.1),
80
  gr.components.Textbox(label="Характеристики",value = example),
81
  ],
82
  outputs=[