Update app.py

app.py

@@ -3,7 +3,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import sqlite3
 import hashlib
-
+import os
 app = FastAPI()
 
 # CORS (allow frontend on another origin like mobile app)
@@ -15,8 +15,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Initialize SQLite
-conn = sqlite3.connect("users.db", check_same_thread=False)
+
+db_path = os.path.join(os.getenv("HF_HOME", "/data"), "users.db")
+conn = sqlite3.connect(db_path, check_same_thread=False)
+
 cursor = conn.cursor()
 cursor.execute('''
     CREATE TABLE IF NOT EXISTS users (