Commit
·
6d2a4c3
1
Parent(s):
e95d03f
deploy
Browse files- services/embedding.py +0 -6
services/embedding.py
CHANGED
|
@@ -5,12 +5,6 @@ from typing import List, Optional
|
|
| 5 |
import torch
|
| 6 |
import os
|
| 7 |
|
| 8 |
-
from huggingface_hub import HfFileSystem
|
| 9 |
-
import transformers.utils.hub
|
| 10 |
-
|
| 11 |
-
# Set cache directory globally for huggingface_hub
|
| 12 |
-
transformers.utils.hub.HUGGINGFACE_HUB_CACHE = "/app/cache/hub"
|
| 13 |
-
|
| 14 |
# Load model
|
| 15 |
model = SentenceTransformer("clip-ViT-B-32")
|
| 16 |
# Load model with custom cache directory
|
|
|
|
| 5 |
import torch
|
| 6 |
import os
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
# Load model
|
| 9 |
model = SentenceTransformer("clip-ViT-B-32")
|
| 10 |
# Load model with custom cache directory
|