Update README.md

README.md

@@ -50,6 +50,22 @@ The following hyperparameters were used during training:
 | 0.7064        | 4.0   | 40   | 0.8390          |
 | 0.6468        | 5.0   | 50   | 0.8028          |
 
+```
+CONVERSION_PROMPT = 'LCT\n'  # LaTeX conversion task
+
+CONVERSION_TOKEN = 'LaTeX:'
+
+loaded_model = GPT2LMHeadModel.from_pretrained('Andyrasika/math_english_to_latex')
+latex_generator = pipeline('text-generation', model=loaded_model, tokenizer=tokenizer)
+
+text_sample = 'r of x is sum from 0 to x of x squared'
+conversion_text_sample = f'{CONVERSION_PROMPT}English: {text_sample}\n{CONVERSION_TOKEN}'
+
+print(latex_generator(
+    conversion_text_sample, num_beams=5, early_stopping=True, temperature=0.7,
+    max_length=len(tokenizer.encode(conversion_text_sample)) + 20
+)[0]['generated_text'])
+```
 
 ### Framework versions