Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def run_tinystyler(source_text, target_example_texts, reranking, temperature, to
|
|
16 |
do_sample=True,
|
17 |
temperature=temperature,
|
18 |
top_p=top_p,
|
19 |
-
max_length=
|
20 |
)
|
21 |
|
22 |
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
@@ -53,7 +53,7 @@ with gr.Blocks(theme="ParityError/[email protected]") as demo:
|
|
53 |
source_text = gr.Textbox(lines=3, placeholder="Enter the source text to transform into the target style...", label="Source Text")
|
54 |
target_example_texts = gr.Textbox(lines=5, placeholder="Enter example texts of the target style (one per line)...", label="Example Texts of the Target Style")
|
55 |
reranking = gr.Slider(1, 10, value=5, step=1, label="Re-ranking")
|
56 |
-
temperature = gr.Slider(0.
|
57 |
top_p = gr.Slider(0.0, 1.0, value=1.0, step=0.1, label="Top-P")
|
58 |
|
59 |
output = gr.Textbox(lines=5, placeholder="Click 'Generate' to transform the source text into the target style.", label="Output", interactive=False)
|
|
|
16 |
do_sample=True,
|
17 |
temperature=temperature,
|
18 |
top_p=top_p,
|
19 |
+
max_length=1024
|
20 |
)
|
21 |
|
22 |
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
|
|
53 |
source_text = gr.Textbox(lines=3, placeholder="Enter the source text to transform into the target style...", label="Source Text")
|
54 |
target_example_texts = gr.Textbox(lines=5, placeholder="Enter example texts of the target style (one per line)...", label="Example Texts of the Target Style")
|
55 |
reranking = gr.Slider(1, 10, value=5, step=1, label="Re-ranking")
|
56 |
+
temperature = gr.Slider(0.1, 2.0, value=1.0, step=0.1, label="Temperature")
|
57 |
top_p = gr.Slider(0.0, 1.0, value=1.0, step=0.1, label="Top-P")
|
58 |
|
59 |
output = gr.Textbox(lines=5, placeholder="Click 'Generate' to transform the source text into the target style.", label="Output", interactive=False)
|