Spaces:
Sleeping
Sleeping
max tokens 128
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ prompt = "### Instruction:\nContinue a dad joke:\n\n### Input:\n{input}\n\n### R
|
|
17 |
|
18 |
def make_response(message, history):
|
19 |
input = prompt.format(input=message)
|
20 |
-
generated = generator(input, return_full_text=False, max_new_tokens=
|
21 |
return generated[0]["generated_text"]
|
22 |
|
23 |
|
|
|
17 |
|
18 |
def make_response(message, history):
|
19 |
input = prompt.format(input=message)
|
20 |
+
generated = generator(input, return_full_text=False, max_new_tokens=128, num_return_sequences=1)
|
21 |
return generated[0]["generated_text"]
|
22 |
|
23 |
|