Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ tags:
|
|
| 35 |
|
| 36 |
## How to Test tinyvvision:
|
| 37 |
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!
|
| 38 |
-
|
| 39 |
import torch, re, numpy as np, math
|
| 40 |
from PIL import Image, ImageDraw, ImageFont
|
| 41 |
|
|
@@ -131,5 +131,5 @@ with torch.no_grad():
|
|
| 131 |
print(" Similarity scores:")
|
| 132 |
for j, c in enumerate(captions):
|
| 133 |
print(f" {c}: {sim[j]:.4f}")
|
| 134 |
-
print(" Best match:", captions[rank], "\n")
|
| 135 |
✨ **Enjoy experimenting!** ✨
|
|
|
|
| 35 |
|
| 36 |
## How to Test tinyvvision:
|
| 37 |
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!
|
| 38 |
+
```from huggingface_hub import hf_hub_download
|
| 39 |
import torch, re, numpy as np, math
|
| 40 |
from PIL import Image, ImageDraw, ImageFont
|
| 41 |
|
|
|
|
| 131 |
print(" Similarity scores:")
|
| 132 |
for j, c in enumerate(captions):
|
| 133 |
print(f" {c}: {sim[j]:.4f}")
|
| 134 |
+
print(" Best match:", captions[rank], "\n")```
|
| 135 |
✨ **Enjoy experimenting!** ✨
|