Update app.py
Browse files
app.py
CHANGED
@@ -39,12 +39,11 @@ def generate_text(prompt):
|
|
39 |
num_beams=5,
|
40 |
repetition_penalty=1.5,
|
41 |
temperature=1,
|
42 |
-
top_k=100,
|
43 |
top_p=0.5,
|
44 |
early_stopping=True
|
45 |
|
46 |
)
|
47 |
-
|
48 |
# Decode the generated tokens
|
49 |
generated_text = tokenizer.decode(outputs, skip_special_tokens=True)[0]
|
50 |
print("show the generated text", generated_text)
|
|
|
39 |
num_beams=5,
|
40 |
repetition_penalty=1.5,
|
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
|
48 |
generated_text = tokenizer.decode(outputs, skip_special_tokens=True)[0]
|
49 |
print("show the generated text", generated_text)
|