Update app.py

app.py

@@ -1,6 +1,7 @@
 from tensorflow.keras.preprocessing.image import ImageDataGenerator
 from tensorflow.keras.applications import MobileNetV2
 from tensorflow.keras import layers, models
+import tensorflow as tf
 
 # Image data generator with augmentation
 train_datagen = ImageDataGenerator(