Update app.py

app.py

@@ -5,10 +5,11 @@ import tensorflow as tf
 model = from_pretrained_keras("araeynn/e")
 
 def image_classifier(inp):
+    class_names = ["Gingivitis", "Hypodontia"]
     inp.save("why.png")
     sunflower_path = "why.png"
     img = tf.keras.utils.load_img(
-      sunflower_path, target_size=(img_height, img_width)
+      sunflower_path, target_size=(180, 180)
     )
     img_array = tf.keras.utils.img_to_array(img)
     img_array = tf.expand_dims(img_array, 0) # Create a batch