mansoorhamidzadeh commited on
Commit
de059de
1 Parent(s): cc4c039

Update README.md

Browse files
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=model, tokenizer=tokenizer)
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