Spaces:
Runtime error
Runtime error
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/Interstellar@0.0.1") 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)
|