Update app.py

app.py

@@ -10,11 +10,13 @@ from pydantic import BaseModel
 from flask import Flask, request, jsonify
 from transformers import pipeline
 from PIL import Image
+import huggingface_hub
 
 # ✅ Set Hugging Face cache to a writable directory
 cache_dir = "/tmp/huggingface_cache"
 os.environ["TRANSFORMERS_CACHE"] = cache_dir
 os.environ["HF_HOME"] = cache_dir
+huggingface_hub.utils.HF_HUB_CACHE = cache_dir  # ✅ Force Hugging Face to use the correct cache directory
 
 # ✅ Ensure the cache directory exists
 os.makedirs(cache_dir, exist_ok=True)