Spaces:

khawir
/

cg-gen-ai

Paused

khawir commited on May 4

Commit

2c304d1

•

1 Parent(s): 67e167f

Update app/config.py

Files changed (1) hide show

app/config.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
-DATABASE_URL = 'sqlite:///data/sql_app.db'
 SECRET_KEY = os.environ.get("SECRET_KEY")
 ALGORITHM = os.environ.get("ALGORITHM")
@@ -11,6 +11,6 @@ ACCESS_TOKEN_EXPIRE_MINUTES = 30000
 SUPERUSER_USERNAME = os.environ.get("SUPERUSER_USERNAME")
 SUPERUSER_PASSWORD = os.environ.get("SUPERUSER_PASSWORD")
-DEVICE = "cuda"
 # os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'

 import os
+DATABASE_URL = os.environ.get("DATABASE_URL")
 SECRET_KEY = os.environ.get("SECRET_KEY")
 ALGORITHM = os.environ.get("ALGORITHM")
 SUPERUSER_USERNAME = os.environ.get("SUPERUSER_USERNAME")
 SUPERUSER_PASSWORD = os.environ.get("SUPERUSER_PASSWORD")
+DEVICE = os.environ.get("DEVICE")
 # os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'