Update app.py

app.py

@@ -1040,52 +1040,52 @@ def create_gradio_interface(model_path: str, default_max_tokens: int = 8192, att
                             label="Play individual segments to verify accuracy"
                         )
         
-        # # Event handlers
-        # do_sample_checkbox.change(
-        #     fn=lambda x: gr.update(visible=x),
-        #     inputs=[do_sample_checkbox],
-        #     outputs=[sampling_params]
-        # )
-        
-        # def reset_stop_flag():
-        #     """Reset stop flag before starting transcription."""
-        #     global stop_generation_flag
-        #     stop_generation_flag = False
-        
-        # def set_stop_flag():
-        #     """Set stop flag to interrupt generation."""
-        #     global stop_generation_flag
-        #     stop_generation_flag = True
-        #     return "⏹️ Stop requested..."
-        
-        # transcribe_button.click(
-        #     fn=reset_stop_flag,
-        #     inputs=[],
-        #     outputs=[],
-        #     queue=False
-        # ).then(
-        #     fn=transcribe_audio,
-        #     inputs=[
-        #         audio_input,
-        #         audio_path_input,
-        #         start_time_input,
-        #         end_time_input,
-        #         max_tokens_slider,
-        #         temperature_slider,
-        #         top_p_slider,
-        #         do_sample_checkbox,
-        #         repetition_penalty_slider,
-        #         context_info_input
-        #     ],
-        #     outputs=[raw_output, audio_segments_output]
-        # )
-        
-        # stop_button.click(
-        #     fn=set_stop_flag,
-        #     inputs=[],
-        #     outputs=[raw_output],
-        #     queue=False
-        # )
+        # Event handlers
+        do_sample_checkbox.change(
+            fn=lambda x: gr.update(visible=x),
+            inputs=[do_sample_checkbox],
+            outputs=[sampling_params]
+        )
+        
+        def reset_stop_flag():
+            """Reset stop flag before starting transcription."""
+            global stop_generation_flag
+            stop_generation_flag = False
+        
+        def set_stop_flag():
+            """Set stop flag to interrupt generation."""
+            global stop_generation_flag
+            stop_generation_flag = True
+            return "⏹️ Stop requested..."
+        
+        transcribe_button.click(
+            fn=reset_stop_flag,
+            inputs=[],
+            outputs=[],
+            queue=False
+        ).then(
+            fn=transcribe_audio,
+            inputs=[
+                audio_input,
+                audio_path_input,
+                start_time_input,
+                end_time_input,
+                max_tokens_slider,
+                temperature_slider,
+                top_p_slider,
+                do_sample_checkbox,
+                repetition_penalty_slider,
+                context_info_input
+            ],
+            outputs=[raw_output, audio_segments_output]
+        )
+        
+        stop_button.click(
+            fn=set_stop_flag,
+            inputs=[],
+            outputs=[raw_output],
+            queue=False
+        )
         
         # Add examples
         gr.Markdown("## 📋 Instructions")
@@ -1137,6 +1137,6 @@ if __name__ == "__main__":
     demo.launch(
         show_error=True,
         theme=gr.themes.Soft(),
-        # css=custom_css
+        css=custom_css
         # 不要加 server_name, server_port, share
     )