readme update #2

README.md

@@ -16,7 +16,7 @@ language:
 
 # kazRush-kk-ru
 
-KazRush is a translation model for translating from Kazakh to Russian.
+KazRush-kk-ru is a translation model for translating from Kazakh to Russian.
 
 ## Usage
 
@@ -36,7 +36,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained('deepvk/kazRush-kk-ru')
 tokenizer = AutoTokenizer.from_pretrained('deepvk/kazRush-kk-ru')
 
 def generate(text, **kwargs):
-    inputs = tokenizer(text, return_tensors='pt')
+    inputs = tokenizer(text, return_tensors='pt').to('cuda')
     with torch.no_grad():
         hypotheses = model.generate(**inputs, num_beams=5, **kwargs)
     return tokenizer.decode(hypotheses[0], skip_special_tokens=True)
@@ -72,7 +72,7 @@ Calculation of BLEU and chrF follows the standart implementation from [sacreBLEU
 | [nllb-200-3.3B](https://huggingface.co/facebook/nllb-200-3.3B)                    | 3.3B    | 21.5  | 50.7  | 0.8874 |
 | [our model (azi7oreu)]()                             | 64.8 M    | 16.3      | 46.6                   | 0.8428 |
 | [our model (fnkx3n1x)]()                             | 64.8 M    | 17.5     | 47.4                   | 0.8029 |
-| [our model (243xhibn)]()                             | 64.8 M    | 17.5     | 47.4                   | 0.8556 |
+| [our model (243xhibn)]()                             | 64.8 M    | 17.4     | 47.4                   | 0.8556 |
 
 ## Examples of usage (ПЕРЕДЕЛАТЬ НА ЛУЧШУЮ, пока что тут примеры с azi7oreu):