Update app.py

app.py

@@ -42,9 +42,8 @@ def list_attributes_and_values():
     """
     return html_script
 
-def retrieve_checkpoint():
-    print(users)
-    return users['save_name']
+def retrieve_checkpoint(user):
+    return user["save_name"]
 
 with gr.Blocks() as demo:
     
@@ -192,7 +191,7 @@ with gr.Blocks() as demo:
 
     #llm
     mist_button.click(chat_with_mistral, inputs=[dd_source_ask, tb_destcol, dd_prompt, fi_excel, dd_url, dd_searchcol, dd_keywords, dd_llm, st_user], outputs=[fi_excel, df_output, dd_prompt, dd_keywords, st_user, fi_config])
-    btn_checkpoint.click(retrieve_checkpoint, outputs=fi_checkpoint)
+    btn_checkpoint.click(retrieve_checkpoint, inputs=st_user, outputs=fi_checkpoint)
     
     #classification
     btn_classif.click(classification, inputs=[dd_source_class, fi_excel, df_category, sl_treshold], outputs=[fi_excel, df_output])