Update app.py

app.py

@@ -74,7 +74,7 @@ def gradio_predict(image):
     predicted_label, confidence = predict(image)
     return f"Predicted Label: {predicted_label}, Confidence: {confidence:.4f}"
 
-def gradio_mask(image, steps, increment):
+def gradio_mask(image, steps, increment=1):
     modified_image, original_label, predicted_label = progressively_mask_image(image, steps, increment)
     return modified_image, f"Original Label: {original_label}, New Label: {predicted_label}"
 
@@ -223,8 +223,8 @@ class GradioInterface:
             with gr.Row(elem_classes="content-container"):
                 with gr.Column():
                     input_image = gr.Image(label="Input Image", type="pil", format="png", elem_classes="image-preview")
-                    steps_input = gr.Slider(minimum=1, maximum=100, label="Steps", step=1, value=100)
-                    increment_input = gr.Slider(minimum=1, maximum=20, label="Increment", step=1, value=5)
+                    steps_input = gr.Slider(minimum=1, maximum=100, label="Attributions Drop Percentage", step=1, value=5)
+                    #increment_input = gr.Slider(minimum=1, maximum=100, label="increment", step=1, value=5)
                     examples = gr.Examples(
                         examples=self.preloaded_examples,
                         inputs=input_image,