Update README.md

README.md

@@ -38,7 +38,7 @@ All the training sets involved in our progressive training scheme that we create
 ## Inference
 
 First install the package following our Github Repo. Then define the model, the renderer, and other utils.
-```
+```python
 import torch
 from PIL import Image
 from pixel import (
@@ -103,8 +103,8 @@ def image_collator(examples):
  }
 ```
 
-Simple code to do inference and similarity computation:
-```
+A minumum example to do inference and similarity computation:
+```python
 texts = ["I love you","I like you"]
 
 inputs = preprocess(texts)