Spaces:

GenAICoder
/

Risk_Manager

Sleeping

GenAICoder commited on May 23

Commit

43a4f8f

verified ·

1 Parent(s): 47e9ea8

Update analytics/ai_assistant.py

Files changed (1) hide show

analytics/ai_assistant.py CHANGED Viewed

@@ -144,7 +144,7 @@ def generate_ai_answer(question: str, df: pd.DataFrame, as_of_month: str = None,
     client = _get_inference_client()
     print("Inference called successfully")
-    response = client.text_generation(
         model=HF_MODEL_ID,
         prompt = prompt,
         max_new_tokens= HF_MAX_TOKENS,

     client = _get_inference_client()
     print("Inference called successfully")
+    response = client.chat.completions.create(
         model=HF_MODEL_ID,
         prompt = prompt,
         max_new_tokens= HF_MAX_TOKENS,