tddf commited on
Commit
1b62453
·
verified ·
1 Parent(s): 9477b8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import AutoProcessor, AutoModelForSeq2SeqLM
8
  # --- Настройка кэша для модели (чтобы не скачивать при каждом запуске) ---
9
  # Hugging Face Spaces даёт нам постоянную папку /data, если включить Persistent Storage.
10
  # Если /data существует, используем её, иначе временную папку /tmp.
11
- CACHE_DIR = "/tmp/.huggingface" if os.path.exists("/data") else "/tmp/.huggingface"
12
  os.makedirs(CACHE_DIR, exist_ok=True)
13
 
14
  # --- Функция загрузки модели с кэшированием через Streamlit ---
 
8
  # --- Настройка кэша для модели (чтобы не скачивать при каждом запуске) ---
9
  # Hugging Face Spaces даёт нам постоянную папку /data, если включить Persistent Storage.
10
  # Если /data существует, используем её, иначе временную папку /tmp.
11
+ CACHE_DIR = "/tmp/.huggingface" if os.path.exists("/data")
12
  os.makedirs(CACHE_DIR, exist_ok=True)
13
 
14
  # --- Функция загрузки модели с кэшированием через Streamlit ---