Update README.md

README.md

@@ -25,18 +25,22 @@ text = "theorem mathd_numbertheory_3 : (∑ x in Finset.range 10, (x + 1) ^ 2) %
 model_inputs = [get_prompt(text)]
 
 print(model_inputs[0])
+```
 
-### Example output : 
-
+Example output:
 
+```lean4
 <theorem>
 theorem mathd_numbertheory_3 : (∑ x in Finset.range 10, (x + 1) ^ 2) % 10 = 5 := by
 </theorem>
 <conjecture>
  theorem mathd_numbertheory_24 : ∑ k in Finset.range 11, k ^ 2 = 385 := by
 </conjecture>
+```
+
+## Inference
 
-### Inference
+```python
 model = LLM(model=model_name, seed=1, trust_remote_code=True, swap_space=8, tensor_parallel_size=1, max_model_len=4096)
 
 sampling_params = SamplingParams(
@@ -48,5 +52,4 @@ sampling_params = SamplingParams(
 
 model_outputs = model.generate(model_inputs, sampling_params, use_tqdm=True)
 print(model_outputs[0].outputs[0].text)
-
-
+```