Spaces:

timm
/

timmAttentionViz

Running

rwightman HF staff commited on Aug 28

Commit

8c53c8a

•

1 Parent(s): cc361dd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -126,7 +126,7 @@ iface = gr.Interface(
         gr.Dropdown(choices=get_attention_models(), label="Select Model")
     ],
     outputs=gr.Gallery(label="Attention Maps"),
-    title="Attention Map Visualizer for timm Models",
     description="Upload an image and select a timm model to visualize its attention maps."
 )

         gr.Dropdown(choices=get_attention_models(), label="Select Model")
     ],
     outputs=gr.Gallery(label="Attention Maps"),
+    title="Attention Map Visualizer for timm Models. NOTE: This is a WIP.",
     description="Upload an image and select a timm model to visualize its attention maps."
 )