Docker deploy

server/caption_server.py

@@ -19,11 +19,14 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+import os
+os.environ['HF_HOME'] = '/cache'
+
 # Initialize the BLIP and GIT models for image captioning
 blip_model_name = "Salesforce/blip-image-captioning-large"
 git_model_name = "microsoft/git-large-coco"
-blip_pipeline = pipeline("image-to-text", model=blip_model_name)
-git_coco_pipeline = pipeline("image-to-text", model=git_model_name)
+blip_pipeline = pipeline("image-to-text", model=blip_model_name , cache_dir='/cache')
+git_coco_pipeline = pipeline("image-to-text", model=git_model_name , cache_dir='/cache')
 
 # Initialize the model for semantic similarity
 model = SentenceTransformer('all-MiniLM-L6-v2')