Update app.py
Browse files
app.py
CHANGED
@@ -36,12 +36,8 @@ def generate_text(prompt):
|
|
36 |
outputs = model.generate(
|
37 |
input_ids = inputs["input_ids"],
|
38 |
max_length=max_length,
|
39 |
-
num_beams=
|
40 |
-
repetition_penalty=
|
41 |
-
temperature=1,
|
42 |
-
top_p=0.5,
|
43 |
-
early_stopping=True
|
44 |
-
|
45 |
)
|
46 |
print("show the output", outputs)
|
47 |
# Decode the generated tokens
|
|
|
36 |
outputs = model.generate(
|
37 |
input_ids = inputs["input_ids"],
|
38 |
max_length=max_length,
|
39 |
+
num_beams=3,
|
40 |
+
repetition_penalty=2.2,
|
|
|
|
|
|
|
|
|
41 |
)
|
42 |
print("show the output", outputs)
|
43 |
# Decode the generated tokens
|