Update app.py

app.py

@@ -4,14 +4,21 @@ from huggingface_hub import InferenceClient
 # Initialize the hugging face inference api client
 client = InferenceClient(
     provider="cerebras",
-    api_key=""
+    api_key="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"
 )
 
 # Define the function to generate stories or poems
 def generate_text(category, theme, tone, length):
     prompt = f"Write a {length} {tone} {category} about {theme}."
-    result = generator(prompt, max_length=80, do_sample=True, temperature=0.7)
-    return result[0]['generated_text']
+
+    try:
+        completion = client.chat.completions.create(
+            model="Qwen/Qwen3-32B",
+            messages=[{"role":"user", "content":prompt}],
+        )
+        return completion.choices[0].message.content
+    except Exception as e:
+        return f"Error: {e}"
 
 # Create Gradio UI
 with gr.Blocks() as demo: