alexfremont commited on
Commit
6f7e5fa
·
1 Parent(s): 5559385

add var d'env

Browse files
Files changed (1) hide show
  1. main.py +5 -0
main.py CHANGED
@@ -21,6 +21,11 @@ app = FastAPI()
21
  image_size = 256
22
  hf_token = os.environ.get("api_read")
23
  VALID_API_KEYS = os.environ.get("api_key")
 
 
 
 
 
24
 
25
 
26
  @app.middleware("http")
 
21
  image_size = 256
22
  hf_token = os.environ.get("api_read")
23
  VALID_API_KEYS = os.environ.get("api_key")
24
+ INSTANCE_GROUP = os.environ.get("INSTANCE_GROUP")
25
+ if INSTANCE_GROUP:
26
+ logging.info(f"INSTANCE_GROUP={INSTANCE_GROUP}")
27
+ else:
28
+ logging.warning("INSTANCE_GROUP not set; all models will be loaded")
29
 
30
 
31
  @app.middleware("http")