Update README.md

README.md

@@ -11,7 +11,9 @@ tags:
 Метрики обучения bleu:100.0 sari:28.699 fkgl:31.931 (из файла "train.logs")
 
 ```
-input_text='Война Советского Союза против фашистской Германии и её союзников (Венгрии, Италии, Румынии, Словакии, Хорватии, Финляндии, Японии); составная часть Второй мировой войны 1939-1945 гг.'
+input_text='''Война Советского Союза против фашистской Германии и её союзников
+              (Венгрии, Италии, Румынии, Словакии, Хорватии, Финляндии, Японии);
+              составная часть Второй мировой войны 1939-1945 гг.'''
 
 def example(source, model, tokenizer):
     """
@@ -24,7 +26,9 @@ def example(source, model, tokenizer):
     print(f'SOURCE: {source}')
     input_ids, attention_mask = tokenizer(source, return_tensors = 'pt').values()
     with torch.no_grad():
-        output = model.generate(input_ids = input_ids.to(model.device), attention_mask = attention_mask.to(model.device), max_new_tokens = input_ids.size(1)*2, min_length=0)
+        output = model.generate(input_ids = input_ids.to(model.device),
+                 attention_mask = attention_mask.to(model.device),
+                 max_new_tokens=input_ids.size(1)*2, min_length=0)
     return tokenizer.decode(output.squeeze(0), skip_special_tokens = True)
 
 example(input_text, model, tokenizer)