Add API key

src/model_demo.py

@@ -12,6 +12,7 @@ from PIL import Image
 from wandb_data import TIMEZONE
 
 SERVER_API = os.environ["SERVER_API"]
+SERVER_API_KEY = os.environ["SERVER_API_KEY"]
 
 current_model: str | None = None
 last_current_model_sync: datetime = datetime.fromtimestamp(0, TIMEZONE)
@@ -50,7 +51,11 @@ def submit(prompt: str, seed: int | str | None, baseline: bool) -> tuple:
         seed = random.randint(0, 2 ** 32 - 1)
 
     print(f"Making request with prompt: {prompt}, seed: {seed}, baseline: {baseline}")
-    response = requests.post(f"{SERVER_API}/generate", params={"prompt": prompt, "baseline": baseline, "seed": seed})
+    response = requests.post(
+        f"{SERVER_API}/generate",
+        params={"prompt": prompt, "baseline": baseline, "seed": seed},
+        headers={"X-API-Key": SERVER_API_KEY},
+    )
     response.raise_for_status()
     result = response.json()
     generation_time = float(result["generation_time"])