Update README.md

README.md

@@ -6,6 +6,7 @@ language:
 - tr
 license: mit
 pipeline_tag: token-classification
+base_model: ytu-ce-cosmos/turkish-base-bert-uncased
 ---
 
 ## Example Usage
@@ -64,9 +65,9 @@ def end2end(sent, capitalization_corr, punc_corr):
 
 
 cap_model = BertForTokenClassification.from_pretrained(
-    "turkish-base-bert-capitalization-correction")
+    "ytu-ce-cosmos/turkish-base-bert-capitalization-correction")
 punc_model = BertForTokenClassification.from_pretrained(
-    "turkish-base-bert-punctuation-correction")
+    "ytu-ce-cosmos/turkish-base-bert-punctuation-correction")
 
 tokenizer = AutoTokenizer.from_pretrained(
     "turkish-base-bert-capitalization-correction")