Update app.py

app.py

@@ -112,18 +112,16 @@ with demo:
             }[dataset]
             filename = f"{model}.json"
             
-            yield f"Downloading sampled reasoning paths from Hugging Face {repo_id}...", gr.update(visible=False), gr.update(), gr.update()
+            yield f"Downloading sampled reasoning paths from Hugging Face {repo_id}...", gr.update(visible=False), gr.update(), gr.update(), gr.update(visible=False)
             file_path = hf_hub_download(repo_id=repo_id, filename=filename, repo_type="dataset")
             
             global json_file
             with open(file_path, 'r', encoding='utf-8') as f:
                 json_file = json.load(f)
             clist = get_available_problems()
-            # yield "Removing downloaded file..."
-            # print(file_path) 
             os.remove(file_path)
-            
-            yield "Loading complete! You can now select a problem ID.", gr.update(visible=True), gr.update(value=f"Dataset: {dataset}\tModel: {model}\tK: {k_value}\tSeed: {seed}"), gr.update(choices=clist, value=clist[0])
+
+            yield "Loading complete! You can now select a problem ID.", gr.update(visible=True), gr.update(value=f"Dataset: {dataset}\tModel: {model}\tK: {k_value}\tSeed: {seed}"), gr.update(choices=clist, value=clist[0]), gr.update(visible=False)
             
         except Exception as e:
             yield f"Error: {str(e)}"
@@ -167,7 +165,7 @@ with demo:
 
         return gr.update(visible=True), gr.update(value=res_to_str(ppl_correct, ppl_answers)), gr.update(value=res_to_str(sc_correct, sc_answers)), gr.update(value=res_to_str(rpc_correct, rpc_answers))
 
-    load_btn.click(fn=load, inputs=[dataset, model, k_value, seed],outputs=[load_btn, content_column, data_info, problem_id], show_progress="inside")
+    load_btn.click(fn=load, inputs=[dataset, model, k_value, seed],outputs=[load_btn, content_column, data_info, problem_id, result_column], show_progress="inside")
     run_btn.click(fn=evaluate, inputs=problem_id, outputs=[result_column, ppl_result, sc_result, rpc_result])
 
 if __name__ == "__main__":