Update app.py

app.py

@@ -566,29 +566,28 @@ with main_block as demo:
             
             def start_evaluation(command, jsonl_file, subset, split):
                 lock.acquire()
-                try:
-                    extra = subset + "_" if subset != "full" else ""
-                    if jsonl_file is not None:
-                        result_path = os.path.basename(jsonl_file.name).replace(".jsonl", f"_{extra}eval_results.json")
+                extra = subset + "_" if subset != "full" else ""
+                if jsonl_file is not None:
+                    result_path = os.path.basename(jsonl_file.name).replace(".jsonl", f"_{extra}eval_results.json")
+                else:
+                    result_path = None
+
+                for log in stream_logs(command, jsonl_file):
+                    if jsonl_file is not None and jsonl_file.name.endswith(".jsonl"):
+                        yield log, gr.update(value=result_path, label=result_path, visible=True), gr.update(visible=False)
                     else:
-                        result_path = None
-
-                    for log in stream_logs(command, jsonl_file):
-                        if jsonl_file is not None and jsonl_file.name.endswith(".jsonl"):
-                            yield log, gr.update(value=result_path, label=result_path, visible=True), gr.update(visible=False)
-                        else:
-                            yield log, gr.update(), gr.update()
-                    result_file = find_result_file()
-                    if result_file:
-                        return gr.update(label="Evaluation completed. Result file found."), gr.update(value=result_file)
-                                # gr.Button(visible=False)#,
-                                # gr.DownloadButton(label="Download Result", value=result_file, visible=True))
-                    else:
-                        return gr.update(label="Evaluation completed. No result file found."), gr.update(value=result_path)
-                                # gr.Button("Run Evaluation", visible=True),
-                                # gr.DownloadButton(visible=False))
-                finally:
-                    lock.release()
+                        yield log, gr.update(), gr.update()
+                lock.release()
+                result_file = find_result_file()
+                if result_file:
+                    return gr.update(label="Evaluation completed. Result file found."), gr.update(value=result_file)
+                            # gr.Button(visible=False)#,
+                            # gr.DownloadButton(label="Download Result", value=result_file, visible=True))
+                else:
+                    return gr.update(label="Evaluation completed. No result file found."), gr.update(value=result_path)
+                            # gr.Button("Run Evaluation", visible=True),
+                            # gr.DownloadButton(visible=False))
+                    
             submit_btn.click(start_evaluation,
                         inputs=[command_output, jsonl_file, subset, split],
                         outputs=[log_output, download_btn, submit_btn])