Spaces:

GenAICoder
/

Risk_Manager

Sleeping

GenAICoder commited on May 23

Commit

3e46fcb

verified ·

1 Parent(s): c1982f2

Update analytics/ai_assistant.py

Files changed (1) hide show

analytics/ai_assistant.py CHANGED Viewed

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

     client = _get_inference_client()
     response = client.text_generation(
         model=HF_MODEL_ID,
+        prompt = prompt,
         max_new_tokens= HF_MAX_TOKENS,
         temperature= HF_TEMPERATURE,
         top_k= 50,