add extra instructions

app.py

@@ -28,8 +28,16 @@ if model_name is not None:
     # Load the model
     model = IndividualLandmarkViT.from_pretrained(model_name, input_size=image_size).eval().to(device)
     num_parts = model.num_landmarks
-    amap_vis = VisualizeAttentionMaps(num_parts=num_parts+1, bg_label=num_parts)
+    amap_vis = VisualizeAttentionMaps(num_parts=num_parts + 1, bg_label=num_parts)
     test_transforms = make_test_transforms(image_size)
+    # Instructions
+    if "cub" or "nabirds" in model_name:
+        st.write("Upload an image of a bird to visualize the attention maps")
+    elif "flowers" in model_name:
+        st.write("Upload an image of a flower to visualize the attention maps")
+    else:
+        st.write("Upload an image of any PartImageNet class (land animals + fish + cars + airplanes) to visualize the attention maps")
+
     image_name = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])  # Upload an image
     if image_name is not None:
         image = Image.open(image_name).convert("RGB")