marianna13 commited on
Commit
56e28b1
1 Parent(s): ebc3409

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -13,7 +13,7 @@ from transformers import pipeline
13
 
14
  max_length = 50
15
  min_length = 10
16
- model_id = "marianna13/flan-t5-small-summarization"
17
 
18
  summarizer = pipeline("summarization", model=model_id, max_length=max_length, min_length=min_length)
19
 
 
13
 
14
  max_length = 50
15
  min_length = 10
16
+ model_id = "marianna13/flan-t5-base-summarization"
17
 
18
  summarizer = pipeline("summarization", model=model_id, max_length=max_length, min_length=min_length)
19