aradhyapavan commited on
Commit
64f8bc6
·
verified ·
1 Parent(s): 15f0f35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- app.config['UPLOAD_FOLDER'] = os.path.join(APP_ROOT, 'uploads')
 
30
  app.config['MAX_CONTENT_LENGTH'] = 50 * 1024 * 1024 # 50MB limit
31
 
32
- # Faiss Configuration
33
- FAISS_INDEX_DIR = os.path.join(APP_ROOT, "faiss_indices")
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