xianghe commited on
Commit
fb82abb
·
1 Parent(s): a9bf99b
Files changed (1) hide show
  1. app.py +19 -24
app.py CHANGED
@@ -35,7 +35,7 @@ def get_model_icon_html(model_name):
35
  model_lower = model_name.lower()
36
  icon_url = None
37
 
38
- # 特殊处理 DS- 开头的模型DeepSeek简称
39
  if model_lower.startswith('ds-'):
40
  icon_url = MODEL_ICON_URLS.get('deepseek')
41
  else:
@@ -514,33 +514,28 @@ with gr.Blocks(title="Jailbreak Attack Results Leaderboard", css="""
514
 
515
  ### Scoring:
516
  Lower scores indicate better resistance to jailbreak attempts.
517
-
518
  ---
519
- *Icons powered by [LobeHub Icons](https://github.com/lobehub/lobe-icons)*
520
  """
521
  )
522
 
523
- gr.Markdown("### 📈 GCG Attack Model Visualization")
524
-
525
- # 显示 PDF 文件
526
- gr.Image(
527
- value="./figs/GCG_attack_model.jpg",
528
- label="GCG Attack Model",
529
- show_label=True,
530
- interactive=False,
531
- container=True
532
- )
533
-
534
- gr.Markdown("### 📈 GCG Defense Model Visualization")
535
-
536
- # 显示 PDF 文件
537
- gr.Image(
538
- value="./figs/GCG_defense_model.jpg",
539
- label="GCG Defense Model",
540
- show_label=True,
541
- interactive=False,
542
- container=True
543
- )
544
 
545
  with gr.Tab("🤖 Model View"):
546
  gr.Markdown("### Compare how models perform against various evaluation methods")
 
35
  model_lower = model_name.lower()
36
  icon_url = None
37
 
38
+ # 特殊处理 DS- 开头的模型(DeepSeek简称)
39
  if model_lower.startswith('ds-'):
40
  icon_url = MODEL_ICON_URLS.get('deepseek')
41
  else:
 
514
 
515
  ### Scoring:
516
  Lower scores indicate better resistance to jailbreak attempts.
 
517
  ---
 
518
  """
519
  )
520
 
521
+ with gr.Row():
522
+ with gr.Column():
523
+ gr.Markdown("### 📈 GCG Attack Model Visualization")
524
+ gr.Image(
525
+ value="./figs/GCG_attack_model.jpg",
526
+ label="GCG Attack Model",
527
+ show_label=False,
528
+ interactive=False
529
+ )
530
+
531
+ with gr.Column():
532
+ gr.Markdown("### 📈 GCG Defense Model Visualization")
533
+ gr.Image(
534
+ value="./figs/GCG_defense_model.jpg",
535
+ label="GCG Defense Model",
536
+ show_label=False,
537
+ interactive=False
538
+ )
 
 
 
539
 
540
  with gr.Tab("🤖 Model View"):
541
  gr.Markdown("### Compare how models perform against various evaluation methods")