Ashkoofaraz commited on
Commit
d0c86df
1 Parent(s): 426b55d

Update README.md

Browse files

Change to text-generation text2text-generation for run pipeline correctly.

Files changed (1) hide show
  1. README.md +1 -1
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="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
 
 
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