Commit ·
631ce73
1
Parent(s): ea1e7dc
fix: change data folder in main
Browse files- 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 |
-
|
|
|
|
| 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)
|