Update app.py

app.py

@@ -6,6 +6,9 @@ model = load_model('myModel.hdf5')
 
 import gradio as gr
 from gradio import inputs
+
+class_names = ['daisy', 'dandelion', 'roses', 'sunflowers', 'tulips']
+
 def classify(img):
   img_array = tf.keras.utils.img_to_array(img)
   img_array = tf.expand_dims(img_array, 0) # Create a batch