Update app/config.py

app/config.py

@@ -1,7 +1,7 @@
 import os
 
 
-DATABASE_URL = 'sqlite:///data/sql_app.db'
+DATABASE_URL = os.environ.get("DATABASE_URL")
 
 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"
+DEVICE = os.environ.get("DEVICE")
 
 # os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'