Spaces:

Chittrarasu
/

image-search-engine-FastAPI

Sleeping

Chittrarasu commited on Mar 2

Commit

f1d9c38

1 Parent(s): 8a14cfe

deploy

Files changed (1) hide show

services/embedding.py CHANGED Viewed

@@ -11,7 +11,7 @@ os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
 os.environ["HF_HOME"] = "/app/cache"
 # Load model with custom cache directory
-model = SentenceTransformer("clip-ViT-B-32", cache_dir="/app/cache")
 def get_text_embedding(text: str) -> Optional[List[float]]:
     try:

 os.environ["HF_HOME"] = "/app/cache"
 # Load model with custom cache directory
+model = SentenceTransformer("clip-ViT-B-32")
 def get_text_embedding(text: str) -> Optional[List[float]]:
     try: