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