Update main.py

main.py

@@ -2,7 +2,7 @@ from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
-import streamlit as st
+import os
 
 app = FastAPI()
 
@@ -55,7 +55,7 @@ def generate(item: Item):
 @app.post("/generate/")
 async def generate_text(item: Item):
     # reject if key not the same 
-    apiKey = st.secrets["API_KEY"]
+    apiKey = os.environ.get("API_KEY")
     # if apiKey != key:
     #     return jsonify({
     #         "image_url": image_url,