commit

app/main.py

@@ -1,3 +1,9 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache/huggingface/transformers"
+os.environ["HF_HOME"] = "/tmp/.cache/huggingface"
+os.makedirs("/tmp/.cache/huggingface/transformers", exist_ok=True)
+
+
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles

app/models/attribute_extractor.py

@@ -1,3 +1,8 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache/huggingface/transformers"
+os.environ["HF_HOME"] = "/tmp/.cache/huggingface"
+os.makedirs("/tmp/.cache/huggingface/transformers", exist_ok=True)
+
 import torch
 from transformers import BlipProcessor, BlipForConditionalGeneration
 from PIL import Image
@@ -6,11 +11,6 @@ import asyncio
 from typing import Dict, Any
 import os 
 
-# Set custom cache directory writable in Spaces environment
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache"
-os.environ["HF_HOME"] = "/tmp/.cache"
-
-
 class AttributeExtractor:
     def __init__(self):
         self.model_name = "Salesforce/blip-image-captioning-base"

app/models/clothing_detector.py

@@ -1,3 +1,8 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache/huggingface/transformers"
+os.environ["HF_HOME"] = "/tmp/.cache/huggingface"
+os.makedirs("/tmp/.cache/huggingface/transformers", exist_ok=True)
+
 import torch
 from sklearn.cluster import KMeans
 import cv2

app/models/color_analyzer.py

@@ -1,3 +1,8 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache/huggingface/transformers"
+os.environ["HF_HOME"] = "/tmp/.cache/huggingface"
+os.makedirs("/tmp/.cache/huggingface/transformers", exist_ok=True)
+
 import cv2
 import numpy as np
 from PIL import Image
@@ -7,9 +12,6 @@ import asyncio
 import webcolors
 import os 
 
-# Set custom cache directory writable in Spaces environment
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache"
-os.environ["HF_HOME"] = "/tmp/.cache"
 
 
 class ColorAnalyzer: