Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-
-client = InferenceClient("deepcogito/cogito-v2-preview-llama-109B-MoE")
+import os
+HF_TOKEN = os.getenv("HF_TOKEN")
+client = InferenceClient("deepcogito/cogito-v2-preview-llama-109B-MoE", HF_TOKEN)
 
 def chat(prompt):
     try:
-        output = client.text_generation(prompt, max_new_tokens=150)
+        output = client.conversational(prompt, max_new_tokens=150)
         return output
     except Exception as e:
         return f"Error: {e}"