add option to change saliency config

app.py

@@ -6,9 +6,6 @@ from matplotlib import pyplot as plt
 import io
 from PIL import Image
 
-# TODO
-# - [ ] Expose more of multi-scale relevancy extractor API
-
 
 def plot_to_png(fig):
     buf = io.BytesIO()
@@ -18,7 +15,7 @@ def plot_to_png(fig):
     return img
 
 
-def generate_relevancy(img: np.array, labels: str, prompt: str):
+def generate_relevancy(img: np.array, labels: str, prompt: str, saliency_config: str):
     labels = labels.split(",")
     prompts = [prompt]
     assert img.dtype == np.uint8
@@ -28,7 +25,7 @@ def generate_relevancy(img: np.array, labels: str, prompt: str):
         img=img,
         text_labels=np.array(labels),
         prompts=prompts,
-        **saliency_configs["ours"](h),
+        **saliency_configs[saliency_config](h),
     )[0]
     grads -= grads.mean(axis=0)
     grads = grads.cpu().numpy()
@@ -59,6 +56,7 @@ iface = gr.Interface(
         gr.Image(type="numpy"),
         gr.Textbox(),
         gr.Textbox(),
+        gr.Dropdown(value="ours", choices=["ours", "chefer_et_al"]),
     ],
     outputs=gr.Image(type="numpy"),
     examples=[