Update app.py

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 model_id = "OpenVINO/Qwen3-0.6B-int4-ov"
 model_path = "Qwen3-0.6B-int4-ov"
 
-hf_hub.snapshot_download(model_id, local_dir=model_path, local_dir_use_symlinks=False) # Added local_dir_use_symlinks=False to avoid potential issues
+hf_hub.snapshot_download(model_id, local_dir=model_path, local_dir_use_symlinks=False)
 
 # 建立推理管線
 device = "CPU"
@@ -31,7 +31,7 @@ def generate_response(prompt):
     output = pipe.generate([prompt], max_length=1024)
     end_time = time.time()
 
-    generated_text = output.text[0]  # Extract the generated text
+    generated_text = output.text[0]
 
     performance_metrics = f"Generate duration: {output.perf_metrics.get_generate_duration().mean:.2f}ms\n"
     performance_metrics += f'Throughput: {output.perf_metrics.get_throughput().mean:.2f} tokens/s'
@@ -45,27 +45,30 @@ def main():
     """
 
     with gr.Blocks() as demo:
-        gr.Markdown("# OpenVINO Qwen3-8B Demo")  # Add a title
+        gr.Markdown("# OpenVINO Qwen3-8B Demo")
         prompt_input = gr.Textbox(lines=3, label="Enter your prompt:")
         output_text = gr.Textbox(label="Generated Response")
-        performance_text = gr.Textbox(label="Performance Metrics", visible=False) # Initially hidden
+        performance_text = gr.Textbox(label="Performance Metrics", visible=False)
+        visibility_state = gr.State(False) # Store visibility state
 
         def update_output(prompt):
             response, performance = generate_response(prompt)
-            return response, performance # return both values
+            return response, performance
 
         prompt_input.change(
             fn=update_output,
             inputs=prompt_input,
-            outputs=[output_text, performance_text],  # Output both response and metrics
+            outputs=[output_text, performance_text],
         )
 
         # Button to show/hide performance metrics
         show_metrics_button = gr.Button("Show/Hide Performance Metrics")
+        def toggle_visibility(visible):
+            return not visible, not visible
         show_metrics_button.click(
-            fn=lambda visible: not visible,
-            inputs=[performance_text.visible],
-            outputs=[performance_text.visible],
+            fn=toggle_visibility,
+            inputs=[visibility_state],
+            outputs=[performance_text.visible, visibility_state],
         )