Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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=[
|