retarfi
commited on
Commit
•
44af809
1
Parent(s):
160c75c
fix wrong argument
Browse files
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
|
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 |
|