AjayP13 commited on
Commit
88976aa
·
verified ·
1 Parent(s): 1b1ed88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=100
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.0, 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)
 
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)