Update README.md

README.md

@@ -35,7 +35,8 @@ tags:
 
 ## How to Test tinyvvision:
 Check out the provided inference script to easily test your own shapes and captions. Feel free to challenge tinyvvision with new, unseen combinations to explore its generalization capability!
-```from huggingface_hub import hf_hub_download
+```python
+from huggingface_hub import hf_hub_download
 import torch, re, numpy as np, math
 from PIL import Image, ImageDraw, ImageFont
 
@@ -131,6 +132,7 @@ with torch.no_grad():
         print("  Similarity scores:")
         for j, c in enumerate(captions):
             print(f"    {c}: {sim[j]:.4f}")
-        print("  Best match:", captions[rank], "\n")```
+        print("  Best match:", captions[rank], "\n")
+```
 
 ✨ **Enjoy experimenting!** ✨