pablo-rf commited on
Commit
a463d62
1 Parent(s): 192ccde

Set few-shot parameters

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -69,8 +69,8 @@ def pass_to_input(generated_gl):
69
 
70
  def fewshot_prompt_parameters():
71
  return (
72
- gr.update(value=int(20)), # max_length
73
- gr.update(value=int(1)) # repetition_penalty
74
  )
75
 
76
  def gradio_app():
@@ -128,7 +128,7 @@ def gradio_app():
128
  label = "Few-shot prompts",
129
  examples = few_shot_prompts_examples,
130
  inputs = [max_length, repetition_penalty],
131
- outputs = [max_length, repetition_penalty],
132
  fn =fewshot_prompt_parameters
133
  )
134
 
 
69
 
70
  def fewshot_prompt_parameters():
71
  return (
72
+ gr.update(value=20), # max_length
73
+ gr.update(value=1) # repetition_penalty
74
  )
75
 
76
  def gradio_app():
 
128
  label = "Few-shot prompts",
129
  examples = few_shot_prompts_examples,
130
  inputs = [max_length, repetition_penalty],
131
+ outputs = [int(max_length), int(repetition_penalty)],
132
  fn =fewshot_prompt_parameters
133
  )
134