Spaces:
Runtime error
Runtime error
Ubuntu
commited on
Commit
·
df71374
1
Parent(s):
5165f6b
update
Browse files
app.py
CHANGED
|
@@ -156,7 +156,7 @@ def main():
|
|
| 156 |
if button_submit:
|
| 157 |
generated_area.text(prompt)
|
| 158 |
report_text = infer(
|
| 159 |
-
|
| 160 |
num_completions=num_completions, seed=seed, stop=literal_eval("'''"+stop+"'''"),
|
| 161 |
)
|
| 162 |
generated_area.text(prompt + report_text)
|
|
|
|
| 156 |
if button_submit:
|
| 157 |
generated_area.text(prompt)
|
| 158 |
report_text = infer(
|
| 159 |
+
prompt, model_name=model_name, max_new_tokens=max_new_tokens, temperature=temperature, top_p=top_p, top_k=top_k,
|
| 160 |
num_completions=num_completions, seed=seed, stop=literal_eval("'''"+stop+"'''"),
|
| 161 |
)
|
| 162 |
generated_area.text(prompt + report_text)
|