Update app.py

app.py

@@ -1,10 +1,9 @@
-import gradio as gr
+import os
 from huggingface_hub import InferenceClient
 
 # Initialize the hugging face inference api client
 client = InferenceClient(
-    provider="cerebras",
-    api_key="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"
+    api_key=os.environ.get("HF_API_TOKEN")
 )
 
 # Define the function to generate stories or poems