Update app.py

app.py

@@ -42,6 +42,8 @@ def list_attributes_and_values():
     """
     return html_script
 
+def retrieve_checkpoint():
+    return "checkpointfile.xlsx"
 
 with gr.Blocks() as demo:
     
@@ -95,6 +97,7 @@ with gr.Blocks() as demo:
                 dd_keywords =  gr.Dropdown(label="Words to look for (Optional)", multiselect=True, allow_custom_value=True, scale=5)
         mist_button = gr.Button("Ask AI")
         fi_checkpoint = gr.File()
+        btn_checkpoint = gr.Button("Retrieve checkpoint file")
             
     with gr.Tab("Classification by topic"):
         gr.Markdown("### This section will categories each contribution in your own personalized categories")
@@ -188,7 +191,8 @@ 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,fi_checkpoint])
-
+    btn_checkpoint.click(retrieve_checkpoint, outputs=fi_checkpoint)
+    
     #classification
     btn_classif.click(classification, inputs=[dd_source_class, fi_excel, df_category, sl_treshold], outputs=[fi_excel, df_output])
     btn_reset_df.click(reset_cate, inputs=df_category, outputs=df_category)