Hugging Face's logo

agentlans
/
flan-t5-small-capitalizer

Safetensors
English
t5
capitalization
Model card Files Community
agentlans commited on
Commit
3bfe011
·
verified ·
1 Parent(s): 3512914

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ import torch
31
  from transformers import pipeline
32
 
33
  device = 0 if torch.cuda.is_available() else -1
34
- model_name = "agentlans/flan-t5-small-capitalizer/"
35
  flan_t5_pipeline = pipeline("text2text-generation", model=model_name, device=device)
36
 
37
  input_text = "buzzfeed's 360-degree look at the aftermath of california's valley fire has been viewed more than 6 million times. plenty of viewers have been asking how we made it."
 
31
  from transformers import pipeline
32
 
33
  device = 0 if torch.cuda.is_available() else -1
34
+ model_name = "agentlans/flan-t5-small-capitalizer"
35
  flan_t5_pipeline = pipeline("text2text-generation", model=model_name, device=device)
36
 
37
  input_text = "buzzfeed's 360-degree look at the aftermath of california's valley fire has been viewed more than 6 million times. plenty of viewers have been asking how we made it."