Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
6f7e5fa
1
Parent(s):
5559385
add var d'env
Browse files
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")
|