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!** ✨
|