Ashkoofaraz
commited on
Commit
•
d0c86df
1
Parent(s):
426b55d
Update README.md
Browse filesChange to text-generation text2text-generation for run pipeline correctly.
README.md
CHANGED
@@ -72,7 +72,7 @@ print(generated_title)
|
|
72 |
|
73 |
|
74 |
# Create a text generation pipeline
|
75 |
-
title_generation_pipeline = pipeline("
|
76 |
generated_title = title_generation_pipeline(input_text, max_length=50, num_beams=5, early_stopping=True)
|
77 |
print(generated_title)
|
78 |
|
|
|
72 |
|
73 |
|
74 |
# Create a text generation pipeline
|
75 |
+
title_generation_pipeline = pipeline("text2text-generation", model="NLPclass/mt5-title-generation")
|
76 |
generated_title = title_generation_pipeline(input_text, max_length=50, num_beams=5, early_stopping=True)
|
77 |
print(generated_title)
|
78 |
|