Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
|
|
| 14 |
|
| 15 |
|
| 16 |
def generate(
|
| 17 |
-
prompt, history,
|
| 18 |
):
|
| 19 |
|
| 20 |
input_text = f"{prompt}, {history}"
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
def generate(
|
| 17 |
+
prompt, history, max_new_tokens=max_new_tokens, repetition_penalty=repetition_penalty, temperature=temperature, top_p=top_p, top_k=top_k
|
| 18 |
):
|
| 19 |
|
| 20 |
input_text = f"{prompt}, {history}"
|