fix
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def generate_prompt(row):
|
|
16 |
return prompt
|
17 |
|
18 |
def test(prompt):
|
19 |
-
outputs = generator(
|
20 |
return outputs[0]["generated_text"]
|
21 |
|
22 |
description = """
|
|
|
16 |
return prompt
|
17 |
|
18 |
def test(prompt):
|
19 |
+
outputs = generator(prompt, max_length=768, num_return_sequences=1, do_sample=True, top_p=0.95, top_k=50)
|
20 |
return outputs[0]["generated_text"]
|
21 |
|
22 |
description = """
|