Update app.py

app.py

@@ -205,7 +205,7 @@ def key_gen_fn() -> Dict:
         key_box: gr.update(visible=True, value=serialized_evaluation_keys_shorten_hex),
         user_id_box: gr.update(visible=True, value=user_id),
         key_len_box: gr.update(
-            visible=True, value=f"{len(serialized_evaluation_keys) / (10**6):.2f} MB"
+            visible=False, value=f"{len(serialized_evaluation_keys) / (10**6):.2f} MB"
         ),
     }
 
@@ -543,7 +543,7 @@ with gr.Blocks() as demo:
     gen_key_btn = gr.Button("Generate the evaluation key")
     error_box2 = gr.Textbox(label="Error ❌", visible=False)
     user_id_box = gr.Textbox(label="User ID:", visible=True)
-    # key_len_box = gr.Textbox(label="Evaluation Key Size:", visible=False)
+    key_len_box = gr.Textbox(label="Evaluation Key Size:", visible=False)
     key_box = gr.Textbox(label="Evaluation key (truncated):", max_lines=3, visible=False)
 
     gen_key_btn.click(
@@ -552,7 +552,7 @@ with gr.Blocks() as demo:
         outputs=[
             key_box,
             user_id_box,
-            # key_len_box,
+            key_len_box,
             error_box2,
         ],
     )