add explicit instructions

app.py

@@ -30,15 +30,15 @@ if model_name is not None:
     num_parts = model.num_landmarks
     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:
-        upload_text = "Upload an image of a bird to visualize the attention maps"
-    elif "flowers" in model_name:
-        upload_text = "Upload an image of a flower to visualize the attention maps"
-    else:
-        upload_text = "Upload an image of any PartImageNet class (land animals + fish + cars + airplanes) to visualize the attention maps"
+    st.write("If you choose to upload an image, the attention map will be displayed.")
+    st.write("The attention map will highlight the regions of the image that the model is focusing on.")
+    st.write("If you choose one of the CUB or NABirds models, please choose a bird image.")
+    st.write("If you choose one of the Flower models, please choose a flower image.")
+    st.write("If you choose one of the PartImageNet models, please choose an images of classes from PartImageNet like land animals/birds/cars/bottles/airplanes.")
 
-    image_name = st.file_uploader(upload_text, type=["jpg", "jpeg", "png"])  # Upload an image
+    image_name = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])  # Upload an image
     if image_name is not None:
         image = Image.open(image_name).convert("RGB")
         image_tensor = test_transforms(image).unsqueeze(0).to(device)