Neobe commited on
Commit
fea6b74
·
verified ·
1 Parent(s): 89f34e1

Update README.md

Browse files

Fixing inference example

Files changed (1) hide show
  1. README.md +7 -4
README.md CHANGED
@@ -48,12 +48,15 @@ It is designed to handle the complexities of journalistic text, focusing on the
48
  ## 💻 Usage
49
 
50
  ```python
51
- from transformers import pipeline
52
 
53
- translator = pipeline("text2text-generation", model="Neobe/dhivehi-byt5-news-transliteration-v1")
 
54
 
55
  text = "ސީރިއާގެ ކަންކަމުގެ ތެރެއަށް އިސްރާއީލް ނުވަނުމަށް ޓްރަމްޕް ގޮވާލައްވައިފި"
56
- result = translator(text, max_length=256)
57
 
58
- print(result[0]['generated_text'])
 
 
59
  # Output: Syria ge kankamuge thereah Israel nuvanumah Trump govaalavvaifi
 
48
  ## 💻 Usage
49
 
50
  ```python
51
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
52
 
53
+ tokenizer = AutoTokenizer.from_pretrained("Neobe/dhivehi-byt5-thaana2latin-v1")
54
+ model = AutoModelForSeq2SeqLM.from_pretrained("Neobe/dhivehi-byt5-thaana2latin-v1")
55
 
56
  text = "ސީރިއާގެ ކަންކަމުގެ ތެރެއަށް އިސްރާއީލް ނުވަނުމަށް ޓްރަމްޕް ގޮވާލައްވައިފި"
57
+ inputs = tokenizer(text, return_tensors="pt")
58
 
59
+ outputs = model.generate(**inputs, max_length=256)
60
+
61
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
62
  # Output: Syria ge kankamuge thereah Israel nuvanumah Trump govaalavvaifi