Update app.py

app.py

@@ -18,20 +18,22 @@ attack = TextFoolerJin2019.build(model_wrapper)
 
 # Function to run attack
 def run_attack(input_text):
-    try:
-        result = attack.attack(input_text, ground_truth_output=1)
+    # Perform the attack
+    result = attack.attack(input_text, ground_truth_output=1)
 
-        skipped_info = ""
-        for goal_function_result in result:
-            if goal_function_result.goal_status == textattack.shared.AttackResultStatus.SKIPPED:
-                skipped_info += f"Skipped Text: {goal_function_result.attacked_text.text}\n"
-
-        model_info = f"Model: {model_name}\n"
-        output = f"{model_info}\nResult:\n{str(result)}\n\nSkipped Details:\n{skipped_info}"
-        return output
-    except Exception as e:
-        return str(e)
+    # Extract detailed information
+    skipped_info = ""
+    for goal_function_result in result:
+        if goal_function_result.goal_status == textattack.shared.AttackResultStatus.SKIPPED:
+            skipped_info += f"Skipped Text: {goal_function_result.attacked_text.text}\n"
+    
+    # Include model information
+    model_info = f"Model: {model_name}\n"
 
+    # Format the output
+    output = f"{model_info}\nResult:\n{str(result)}\n\nSkipped Details:\n{skipped_info}"
+    return output
+    
 # Gradio UI
 gr.Interface(fn=run_attack, 
              inputs=gr.Textbox(lines=4, placeholder="Enter sentence to attack..."),