Update app.py
Browse files
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")
|
| 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 ---
|