| from multilingual_clip.legacy_multilingual_clip import MultilingualClip | |
| model_path = 'M-CLIP/Swedish-500k' | |
| tok_path = 'M-CLIP/Swedish-500k' | |
| head_weight_path = 'data/weights/Swe-CLIP Linear Weights.pkl' | |
| sweclip_args = {'model_name': model_path, | |
| 'tokenizer_name': tok_path, | |
| 'head_path': head_weight_path} | |
| sweclip = MultilingualClip(**sweclip_args) | |
| print(sweclip('test')) |