logan-codes commited on
Commit
631ce73
·
1 Parent(s): ea1e7dc

fix: change data folder in main

Browse files
Files changed (1) hide show
  1. app/main.py +2 -1
app/main.py CHANGED
@@ -116,7 +116,8 @@ async def health_check():
116
 
117
  @app.post("/document")
118
  async def upload_file(background_tasks: BackgroundTasks,file:UploadFile=File(...) ):
119
- upload_dir = os.path.join(os.getenv("DATA_DIR"), "uploads")
 
120
  os.makedirs(upload_dir, exist_ok=True)
121
 
122
  safe_filename = secure_filename(file.filename)
 
116
 
117
  @app.post("/document")
118
  async def upload_file(background_tasks: BackgroundTasks,file:UploadFile=File(...) ):
119
+ base_dir = os.getenv("HF_HOME", "/home/user/app")
120
+ upload_dir = os.path.join(base_dir, "data")
121
  os.makedirs(upload_dir, exist_ok=True)
122
 
123
  safe_filename = secure_filename(file.filename)