Update app.py

app.py

@@ -6,12 +6,15 @@ import tensorflow as tf
 from tensorflow.keras.applications.resnet50 import preprocess_input 
 from tensorflow.keras.preprocessing.image import img_to_array    
 
+from datasets import load_dataset
+
 def main():
     st.title('Jacaranda Identification')
     
     st.markdown("This is a Deep Learning application to identify if a satellite image clip contains Jacaranda trees.\n")
     st.markdown('The predicting result will be "Jacaranda", or "Others".')
     st.markdown('You can click "Browse files" multiple times until adding all images before generating prediction.\n')
+    st.markdown('The image clips can look like these examples. The image size can be arbitrary but better not include too much contents.')
 
     run_the_app()
     
@@ -42,6 +45,9 @@ def convert_df(df):
 
 def run_the_app():
 
+    dataset = load_dataset('jacaranda', split='train')
+    st.image(dataset[0:5]['image'])
+
     class_names = ['Jacaranda', 'Others']
     model = load_model()
     df = generate_df()