Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,6 +5,19 @@ import io
 import torch
 from transformers import CLIPProcessor, CLIPModel
 import os
+import os
+import streamlit as st
+from PIL import Image
+import torch
+from transformers import CLIPProcessor, CLIPModel
+import io
+
+# ✅ Fix for Hugging Face Spaces: use writable /tmp directory
+cache_dir = "/tmp/models"
+os.environ["TRANSFORMERS_CACHE"] = cache_dir
+os.environ["HF_HOME"] = cache_dir
+os.makedirs(cache_dir, exist_ok=True)
+
 
 # Set a writable cache directory (inside the app folder)
 os.environ["TRANSFORMERS_CACHE"] = "./models"