Spaces:
Build error
Build 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}"
|