retarfi commited on
Commit
44af809
1 Parent(s): 160c75c

fix wrong argument

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -183,6 +183,7 @@ def evaluate(
183
  "top_k": top_k,
184
  "num_beams": num_beams,
185
  "max_tokens": max_tokens,
 
186
  },
187
  )
188
  except Exception as e:
@@ -233,6 +234,7 @@ def evaluate(
233
  "top_k": top_k,
234
  "num_beams": num_beams,
235
  "max_tokens": max_tokens,
 
236
  },
237
  )
238
  except Exception as e:
@@ -353,8 +355,8 @@ with gr.Blocks(
353
  submit_button.click(no_interactive, [], [submit_button, clear_button])
354
  submit_button.click(
355
  evaluate,
356
- [instruction, inputs, temperature, max_tokens],
357
- [outputs, submit_button, clear_button, repetition_penalty],
358
  )
359
  clear_button.click(reset_textbox, [], [instruction, inputs, outputs], queue=False)
360
 
 
183
  "top_k": top_k,
184
  "num_beams": num_beams,
185
  "max_tokens": max_tokens,
186
+ "repetition_penalty": repetition_penalty,
187
  },
188
  )
189
  except Exception as e:
 
234
  "top_k": top_k,
235
  "num_beams": num_beams,
236
  "max_tokens": max_tokens,
237
+ "repetition_penalty": repetition_penalty,
238
  },
239
  )
240
  except Exception as e:
 
355
  submit_button.click(no_interactive, [], [submit_button, clear_button])
356
  submit_button.click(
357
  evaluate,
358
+ [instruction, inputs, temperature, max_tokens, repetition_penalty],
359
+ [outputs, submit_button, clear_button],
360
  )
361
  clear_button.click(reset_textbox, [], [instruction, inputs, outputs], queue=False)
362