yujiepan commited on
Commit
9f76d30
Β·
1 Parent(s): ffc03fb

Reorganize layout: inputs and buttons on left, keys list on right

Browse files
Files changed (1) hide show
  1. app.py +21 -22
app.py CHANGED
@@ -293,28 +293,27 @@ with gr.Blocks(title="HuggingFace Model Weight Inspector") as demo:
293
  gr.Markdown("# πŸ” HuggingFace Model Weight Inspector")
294
 
295
  with gr.Row():
296
- model_id_input = gr.Textbox(
297
- label="Model ID",
298
- placeholder="e.g., meta-llama/Llama-2-7b-hf",
299
- value="zai-org/GLM-5",
300
- scale=4,
301
- )
302
- list_keys_btn = gr.Button("πŸ“‹ List Keys", variant="secondary", scale=1)
303
-
304
- keys_output = gr.Textbox(
305
- label="Available Parameter Keys",
306
- lines=3,
307
- max_lines=8,
308
- )
309
-
310
- with gr.Row():
311
- param_key_input = gr.Textbox(
312
- label="Parameter Key",
313
- placeholder="e.g., model.embed_tokens.weight",
314
- value="model.layers.5.mlp.gate.e_score_correction_bias",
315
- scale=4,
316
- )
317
- fetch_btn = gr.Button("πŸ”Ž Fetch", variant="primary", scale=1)
318
 
319
  with gr.Tabs():
320
  with gr.Tab("Results"):
 
293
  gr.Markdown("# πŸ” HuggingFace Model Weight Inspector")
294
 
295
  with gr.Row():
296
+ with gr.Column(scale=1):
297
+ model_id_input = gr.Textbox(
298
+ label="Model ID",
299
+ placeholder="e.g., meta-llama/Llama-2-7b-hf",
300
+ value="Qwen/Qwen3-Coder-Next-FP8",
301
+ )
302
+ param_key_input = gr.Textbox(
303
+ label="Parameter Key",
304
+ placeholder="e.g., model.norm.weight",
305
+ value="model.norm.weight",
306
+ )
307
+ with gr.Row():
308
+ list_keys_btn = gr.Button("πŸ“‹ List Keys", variant="secondary", scale=1)
309
+ fetch_btn = gr.Button("πŸ”Ž Fetch", variant="primary", scale=1)
310
+
311
+ with gr.Column(scale=1):
312
+ keys_output = gr.Textbox(
313
+ label="Available Parameter Keys",
314
+ lines=10,
315
+ max_lines=20,
316
+ )
 
317
 
318
  with gr.Tabs():
319
  with gr.Tab("Results"):