Update app.py

app.py

@@ -5,11 +5,19 @@ import torch
 from PIL import Image
 from transformers import AutoProcessor, AutoModelForSeq2SeqLM
 
-# --- Настройка кэша для модели (чтобы не скачивать при каждом запуске) ---
-# Hugging Face Spaces даёт нам постоянную папку /data, если включить Persistent Storage.
-# Если /data существует, используем её, иначе временную папку /tmp.
-CACHE_DIR = "/tmp/.huggingface" if os.path.exists("/data")
+import os
+import tempfile
+
+# Определяем папку для кэша: /data если есть (persistent), иначе /tmp
+if os.path.exists("/data") and os.access("/data", os.W_OK):
+    CACHE_DIR = "/data/.huggingface"
+else:
+    CACHE_DIR = os.path.join(tempfile.gettempdir(), ".huggingface")
+
 os.makedirs(CACHE_DIR, exist_ok=True)
+os.environ["HF_HOME"] = CACHE_DIR
+os.environ["HF_HUB_CACHE"] = os.path.join(CACHE_DIR, "hub")
+os.environ["TRANSFORMERS_CACHE"] = os.path.join(CACHE_DIR, "transformers")
 
 # --- Функция загрузки модели с кэшированием через Streamlit ---
 @st.cache_resource  # Эта декорация сохраняет модель в памяти между запусками