update

README.md

@@ -4,7 +4,7 @@ emoji: 🌍
 colorFrom: gray
 colorTo: indigo
 sdk: gradio
-sdk_version: 3.50.2
+sdk_version: 4.44.1
 app_file: app.py
 pinned: false
 license: mit
@@ -18,4 +18,4 @@ short_description: XTransferBench Demo for UAP Linf
 
 Demo for ICML2025 paper ["X-Transfer Attacks: Towards Super Transferable Adversarial Attacks on CLIP"](https://arxiv.org/abs/2505.05528)
 
-Code: https://github.com/HanxunH/XTransferBench
+Code: https://github.com/HanxunH/XTransferBench

app.py

@@ -17,25 +17,28 @@ def XTransferAttack(image, attacker_name, epsilon):
     adv_image= adv_image_resized.resize((original_image.size[0], original_image.size[1]))
     return adv_image_resized, adv_image
 
-
 demo = gr.Interface(
     fn=XTransferAttack,
     inputs=[
         gr.Image(type="pil", label="Input Image",),
         gr.Dropdown(
-            XTransferBench.zoo.list_attacker(threat_model='linf_non_targeted'), value="xtransfer_large_linf_eps12_non_targeted", 
-            multiselect=None, label="Attacker", 
-            info="Please select an attacker from the list."
+            choices=XTransferBench.zoo.list_attacker(threat_model='linf_non_targeted'),
+            value="xtransfer_large_linf_eps12_non_targeted",
+            label="Attacker",
+            info="Select an attacker method."
         ),
         gr.Slider(
-            minimum=0, maximum=255, step=1, value=12, label="Epsilon", 
-            info="Maximum perturbation allowed for the attack. The value should be between 0 and 255."
-        ),
+            minimum=0, maximum=255, step=1, value=12,
+            label="Epsilon",
+            info="Perturbation strength (0-255)"
+        )
     ],
     outputs=[
         gr.Image(type="pil", label="Adversarial Image (Resized)"), 
         gr.Image(type="pil", label="Adversarial Image (Original Size)"), 
-    ]
+    ],
+    title="XTransferBench Adversarial Attack Demo",
+    description="Visualize adversarial examples on images using transfer-based attacks from the XTransferBench library.",
 )
 
 demo.launch(share=True)

requirements.txt

@@ -2,4 +2,5 @@ torch
 torchvision
 huggingface-hub
 safetensors
-transformers
+transformers
+XTransferBench