set max labels length

app.py

@@ -39,6 +39,8 @@ def generate_relevancy(
     img: np.array, labels: str, prompt: str, saliency_config: str, subtract_mean: bool
 ):
     labels = labels.split(",")
+    if len(labels) > 32:
+        labels = labels[:32]
     prompts = [prompt]
     img = np.asarray(Image.fromarray(img).resize((244 * 4, 244 * 4)))
     assert img.dtype == np.uint8
@@ -87,8 +89,10 @@ This relevancy extractor builds heavily on [Chefer et al.'s codebase](https://gi
     cache_examples=True,
     inputs=[
         gr.Image(type="numpy", label="Image"),
-        gr.Textbox(label="Labels (comma separated)"),
-        gr.Textbox(label="Prompt"),
+        gr.Textbox(label="Labels (comma separated without spaces in between)"),
+        gr.Textbox(
+            label="Prompt. (Make sure to include \{\} in the prompt like examples below)"
+        ),
         gr.Dropdown(
             value="ours",
             choices=["ours", "ours_fast", "chefer_et_al"],
@@ -128,5 +132,4 @@ This relevancy extractor builds heavily on [Chefer et al.'s codebase](https://gi
         ],
     ],
 )
-# iface.launch(share=True)
 iface.launch()