ifweoi

README.md

@@ -28,7 +28,7 @@ The above uses the default variant, `bort-pr-sp-noisy`. Each variant from the pa
 like so:
 
 ```python
-BartForConditionalGeneration.from_pretrained("rcgale/bort-test", variant="bort-sp")
+BartForConditionalGeneration.from_pretrained("palat/bort", variant="bort-sp")
 ```
 
 The following variants are available, pre-trained on the specified proportion of each task:
@@ -55,27 +55,16 @@ from combining phonemes.)
 ```python
 from transformers import AutoTokenizer, BartForConditionalGeneration
 
-tokenizer = AutoTokenizer.from_pretrained("rcgale/bort-test")
-model = BartForConditionalGeneration.from_pretrained("rcgale/bort-test")
-
 in_texts = [
     "Due to its coastal location, Long ·aɪlən·d winter temperatures are milder than most of the state.",
-    # Output:
-    # Due to its coastal location, Long Island winter temperatures are milder than most of the state.",
-    
     "Due to its coastal location, Long ·b·iʧ winter temperatures are milder than most of the state.",
-    # Output:
-    # Due to its coastal location, Long Beach winter temperatures are milder than most of the state.",
-
     "Due to its coastal location, lɔŋ ·aɪlən·d winter temperatures are milder than most of the state.",
-    # Output:
-    # Due to its coastal location, Long Island winter temperatures are milder than most of the state.",
-
     "Due to its coastal location, lɔŋ ·b·iʧ winter temperatures are milder than most of the state.",
-    # Output:
-    # Due to its coastal location, long beech winter temperatures are milder than most of the state.",
 ]
 
+tokenizer = AutoTokenizer.from_pretrained("palat/bort")
+model = BartForConditionalGeneration.from_pretrained("palat/bort")
+
 inputs = tokenizer(in_texts, return_tensors="pt", padding=True)
 summary_ids = model.generate(inputs["input_ids"], num_beams=2, min_length=0, max_length=2048)
 decoded = tokenizer.batch_decode(summary_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)
@@ -85,6 +74,23 @@ for in_text, out_text in zip(in_texts, decoded):
     print(f"Out:  \t{out_text}")
     print()
 ```
+
+Full output for the above example:
+```
+In:   	Due to its coastal location, lɔŋ ·aɪlən·d winter temperatures are milder than most of the state.
+Out:  	Due to its coastal location, Long Island winter temperatures are milder than most of the state.
+
+In:   	Due to its coastal location, lɔŋ ·b·iʧ winter temperatures are milder than most of the state.
+Out:  	Due to its coastal location, long beech winter temperatures are milder than most of the state.
+
+In:   	Due to its coastal location, Long ·b·iʧ winter temperatures are milder than most of the state.
+Out:  	Due to its coastal location, Long Beach winter temperatures are milder than most of the state.
+
+In:   	Due to its coastal location, lɔŋfɝd winter temperatures are milder than most of the state.
+Out:  	Due to its coastal location, Longford winter temperatures are milder than most of the state.
+```
+
+
 ## Wikipedia Dataset Used in Pre-Training
 
 The BPE-tokenized version of the dataset, including metadata used in word transforms.