try different text based on model choice

app.py

@@ -32,13 +32,13 @@ if model_name is not None:
     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")
+        upload_text = "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")
+        upload_text = "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")
+        upload_text = "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
+    image_name = st.file_uploader(upload_text, 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)