Update analytics/ai_assistant.py

analytics/ai_assistant.py

@@ -141,7 +141,7 @@ def generate_ai_answer(question: str, df: pd.DataFrame, as_of_month: str = None,
     )
 
     client = _get_inference_client()
-    response = client(prompt, parameters={
+    response = client.text_generation(prompt, parameters={
         "max_new_tokens": HF_MAX_TOKENS,
         "temperature": HF_TEMPERATURE,
         "top_k": 50,