NLPclass
/

mt5-title-generation

Text Generation

text2text-generation

title-generation

Inference Endpoints

Model card Files Files and versions Community

Update README.md

#1

by Ashkoofaraz - opened 10 days ago

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -72,7 +72,7 @@ print(generated_title)
 # Create a text generation pipeline
-title_generation_pipeline = pipeline("text-generation", model="NLPclass/mt5-title-generation")
 generated_title = title_generation_pipeline(input_text, max_length=50, num_beams=5, early_stopping=True)
 print(generated_title)

 # Create a text generation pipeline
+title_generation_pipeline = pipeline("text2text-generation", model="NLPclass/mt5-title-generation")
 generated_title = title_generation_pipeline(input_text, max_length=50, num_beams=5, early_stopping=True)
 print(generated_title)