Update main.py

main.py

@@ -4,7 +4,7 @@ from pydantic import BaseModel  # Import BaseModel from pydantic for data valida
 from huggingface_hub import InferenceClient  # Import InferenceClient from huggingface_hub
 import uvicorn  # Import uvicorn for running the FastAPI application
 
-app = FastAPI(HF_API_TOKEN)  # Create a FastAPI instance
+app = FastAPI()  # Create a FastAPI instance
 
 # Define the primary and fallback models
 primary = "mistralai/Mixtral-8x7B-Instruct-v0.1"
@@ -126,7 +126,7 @@ async def generate_text(item: Item = None):
         raise HTTPException(status_code=500, detail=error)
 
     if "KEY" in os.environ:
-        if item.key != os.environ["KEY"]:
+        if item.key != os.environ["HF_API_TOKEN"]:
             raise HTTPException(status_code=401, detail="Valid key is required.")
 
 if __name__ == "__main__":