Update app.py

app.py

@@ -27,9 +27,9 @@ constraints = (
 def generate_image(prompt):
     print("Generating image with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
-    token = os.getenv("API_TOKEN")  # Fetch from environment
+    token = os.getenv("HF_TOKEN")  # Fetch from environment
     if not token:
-        raise ValueError("API_TOKEN environment variable is not set.")
+        raise ValueError("HF_TOKEN environment variable is not set.")
     headers = {
         "Authorization": f"Bearer {token}"
     }