teamapocalypseml
/

regben2ipa-byt5small

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

smji commited on Mar 14

Commit

1e38ea9

•

1 Parent(s): bb74e38

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -51,8 +51,8 @@ Supported district tokens:
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-tokenizer = AutoTokenizer.from_pretrained("smji/ben2ipa-byt5small")
-model = AutoModelForSeq2SeqLM.from_pretrained("smji/ben2ipa-byt5small")
 """
   The format of the input text MUST BE: <district> <bengali_text>
@@ -70,7 +70,7 @@ from transformers import pipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = pipeline("text2text-generation", model="smji/ben2ipa-byt5small", device=device)
 """

 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("teamapocalypseml/ben2ipa-byt5small")
+model = AutoModelForSeq2SeqLM.from_pretrained("teamapocalypseml/ben2ipa-byt5small")
 """
   The format of the input text MUST BE: <district> <bengali_text>
 device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = pipeline("text2text-generation", model="teamapocalypseml/ben2ipa-byt5small", device=device)
 """