Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ def generate_text(prompt):
|
|
45 |
)
|
46 |
print("show the output", outputs)
|
47 |
# Decode the generated tokens
|
48 |
-
generated_text = tokenizer.decode(outputs, skip_special_tokens=True)
|
49 |
print("show the generated text", generated_text)
|
50 |
return generated_text
|
51 |
#############
|
|
|
45 |
)
|
46 |
print("show the output", outputs)
|
47 |
# Decode the generated tokens
|
48 |
+
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
49 |
print("show the generated text", generated_text)
|
50 |
return generated_text
|
51 |
#############
|