UBC-NLP
/

AraT5v2-base-1024

Arabic Machine Translation

Arabic Text Summarization

Arabic News Title and Question Generation

Arabic Paraphrasing and Transliteration

Arabic Code-Switched Translation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Tawkat commited on Aug 16, 2023

Commit

a493377

·

1 Parent(s): d4f87e1

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -32,10 +32,10 @@ from transformers import T5Tokenizer, AutoModelForSeq2SeqLM
 tokenizer = T5Tokenizer.from_pretrained("UBC-NLP/AraT5v2-base-1024")
 model = AutoModelForSeq2SeqLM.from_pretrained("UBC-NLP/AraT5v2-base-1024")
-prompt="عاصمة ألمانيا هي <extra_id_0> "
 input_ids = tokenizer(ar_prompt, return_tensors="pt").input_ids
 outputs = model.generate(input_ids)
-print("Tokenized input:", tokenizer.tokenize(prompt))
 print("Decoded output:", tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```

 tokenizer = T5Tokenizer.from_pretrained("UBC-NLP/AraT5v2-base-1024")
 model = AutoModelForSeq2SeqLM.from_pretrained("UBC-NLP/AraT5v2-base-1024")
+ar_prompt="عاصمة ألمانيا هي <extra_id_0> "
 input_ids = tokenizer(ar_prompt, return_tensors="pt").input_ids
 outputs = model.generate(input_ids)
+print("Tokenized input:", tokenizer.tokenize(ar_prompt))
 print("Decoded output:", tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```