Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,11 +26,12 @@ load_dotenv()
|
|
| 26 |
app = Flask(__name__)
|
| 27 |
app.secret_key = os.getenv("SECRET_KEY", "default_secret_key")
|
| 28 |
APP_ROOT = os.path.abspath(os.path.dirname(__file__))
|
| 29 |
-
|
|
|
|
| 30 |
app.config['MAX_CONTENT_LENGTH'] = 50 * 1024 * 1024 # 50MB limit
|
| 31 |
|
| 32 |
-
# Faiss Configuration
|
| 33 |
-
FAISS_INDEX_DIR = os.path.join(
|
| 34 |
EMBEDDING_DIM = 768 # all-mpnet-base-v2 embedding dimension
|
| 35 |
|
| 36 |
# Ensure writable directories exist when running under gunicorn
|
|
|
|
| 26 |
app = Flask(__name__)
|
| 27 |
app.secret_key = os.getenv("SECRET_KEY", "default_secret_key")
|
| 28 |
APP_ROOT = os.path.abspath(os.path.dirname(__file__))
|
| 29 |
+
DATA_ROOT = os.getenv('HF_DATA_DIR', '/data')
|
| 30 |
+
app.config['UPLOAD_FOLDER'] = os.path.join(DATA_ROOT, 'uploads')
|
| 31 |
app.config['MAX_CONTENT_LENGTH'] = 50 * 1024 * 1024 # 50MB limit
|
| 32 |
|
| 33 |
+
# Faiss Configuration (store under /data for HF Spaces persistence)
|
| 34 |
+
FAISS_INDEX_DIR = os.path.join(DATA_ROOT, "faiss_indices")
|
| 35 |
EMBEDDING_DIM = 768 # all-mpnet-base-v2 embedding dimension
|
| 36 |
|
| 37 |
# Ensure writable directories exist when running under gunicorn
|