Refine UI: smaller font for tensor preview and compact file download component

app.py

@@ -195,8 +195,8 @@ def fetch_param(model_id: str, param_key: str, progress=gr.Progress()):
         preview_str = f"**First {preview_size} values:**\n```\n" + \
             "\n".join(preview_lines) + "\n```"
 
-        if flat.numel() > preview_size:
-            preview_str += f"\n\n... and {flat.numel() - preview_size:,} more values"
+        # if flat.numel() > preview_size:
+        #     preview_str += f"\n\n... and {flat.numel() - preview_size:,} more values"
 
         # Save tensor for download
         log_buffer.write(f"💾 Saving tensor for download...\n")
@@ -331,6 +331,16 @@ custom_css = """
 .compact-row {
     gap: 0.5rem !important;
 }
+.tensor-preview pre {
+    font-size: 0.75rem !important;
+    line-height: 1.0 !important;
+}
+.compact-file {
+    max-height: 80px !important;
+}
+.compact-file > div {
+    min-height: 60px !important;
+}
 """
 
 with gr.Blocks(title="Hugging Face Model Weight Inspector") as demo:
@@ -364,10 +374,10 @@ with gr.Blocks(title="Hugging Face Model Weight Inspector") as demo:
         with gr.Tab("Results"):
             with gr.Row():
                 with gr.Column():
-                    info_output = gr.Markdown(label="Tensor Info")
+                    preview_output = gr.Markdown(label="Tensor Preview", elem_classes="tensor-preview")
                 with gr.Column():
-                    preview_output = gr.Markdown(label="Tensor Preview")
-            download_output = gr.File(label="Download Tensor (.pt file)")
+                    info_output = gr.Markdown(label="Tensor Info")
+            download_output = gr.File(label="Download Tensor (.pt file)", elem_classes="compact-file")
             log_output = gr.Textbox(
                 label="📋 Download Log", lines=1, interactive=False)