Update src/insurance_assistants/ui.py

src/insurance_assistants/ui.py

@@ -10,6 +10,7 @@ from gradio_pdf import PDF
 from smolagents.gradio_ui import _process_action_step, _process_final_answer_step
 from smolagents.memory import ActionStep, FinalAnswerStep, MemoryStep, PlanningStep
 from smolagents.models import ChatMessageStreamDelta
+from huggingface_hub import login
 
 # from smolagents import CodeAgent, InferenceClientModel
 from src.insurance_assistants.agents import manager_agent
@@ -224,6 +225,7 @@ class UI:
                         with gr.Group():
                             api_key = gr.Textbox(placeholder="Enter your HuggingFace Inference API KEY HERE", 
                                                  label="API key",show_label=True, type='password')
+                            login(token=api_key)
                             gr.Markdown(
                                 value="**Your question, please...**", container=True
                             )