MrTechie commited on
Commit
e1292ea
·
1 Parent(s): 01f935e

DB updated for HF

Browse files
Files changed (1) hide show
  1. api/database.py +4 -1
api/database.py CHANGED
@@ -8,7 +8,10 @@ from dotenv import load_dotenv
8
 
9
  load_dotenv()
10
 
11
- DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./test.db")
 
 
 
12
 
13
  engine = create_engine(
14
  DATABASE_URL, connect_args={"check_same_thread": False} if DATABASE_URL.startswith("sqlite") else {}
 
8
 
9
  load_dotenv()
10
 
11
+ DATABASE_URL = os.getenv(
12
+ "DATABASE_URL",
13
+ "sqlite:////data/test.db" if os.getenv("SPACE_ID") else "sqlite:///./test.db"
14
+ ) if os.getenv("SPACE_ID") else "sqlite:///./test.db"
15
 
16
  engine = create_engine(
17
  DATABASE_URL, connect_args={"check_same_thread": False} if DATABASE_URL.startswith("sqlite") else {}