Updated usage example

README.md

@@ -23,6 +23,7 @@ Usage example:
 ---
 
 ``` python
+import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from IPython.display import display, Math, Latex
 
@@ -36,16 +37,16 @@ model.to(device)
 def get_latex(text):
   inputs = tokenizer(text, return_tensors='pt').to(device)
   with torch.no_grad():
-  hypotheses = model.generate(
-  **inputs,
-  do_sample=True, num_return_sequences=1,
-  repetition_penalty=1.2,
-  max_length=len(text),
-  num_beams=10,
-  early_stopping=True
-  )
+    hypotheses = model.generate(
+    **inputs,
+    do_sample=True, num_return_sequences=1,
+    repetition_penalty=1.2,
+    max_length=len(text),
+    num_beams=10,
+    early_stopping=True
+    )
   for h in hypotheses:
-  display(Latex(tokenizer.decode(h, skip_special_tokens=True)))
+    display(Latex(tokenizer.decode(h, skip_special_tokens=True)))
 
 text = '''лямбда прописная квадрат минус три равно десять игрек куб
         При этом шинус икс равен интеграл от экспоненты до трёх игрек штрих'''