mansoorhamidzadeh
commited on
Commit
•
de059de
1
Parent(s):
cc4c039
Update README.md
Browse files
README.md
CHANGED
@@ -72,7 +72,7 @@ print(generated_title)
|
|
72 |
|
73 |
|
74 |
# Create a text generation pipeline
|
75 |
-
title_generation_pipeline = pipeline("text-generation", model=
|
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("text-generation", model="mansoorhamidzadeh/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 |
|