Update app.py

app.py

@@ -31,7 +31,12 @@ HF_HUB_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 if not HF_HUB_TOKEN:
     raise ValueError("Missing Hugging Face API token. Please set HUGGINGFACEHUB_API_TOKEN.")
 
-print(f"HUGGINGFACEHUB_API_TOKEN: {HUGGINGFACEHUB_API_TOKEN}")
+# Get Hugging Face API token from environment variable
+HF_SERVERLESS_API_TOKEN = os.getenv("HF_TOKEN_SERVERLESS_API")
+if not HF_SERVERLESS_API_TOKEN:
+    raise ValueError("Missing HF SERVERLESS API token. Please set HF_TOKEN_SERVERLESS_API.")
+
+print(f"HF_TOKEN_SERVERLESS_API: {HF_SERVERLESS_API_TOKEN}")
 print(f"HF_HUB_TOKEN: {HF_HUB_TOKEN}")
 
 # Login and initialize the InferenceClient