Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 import tensorflow as tf
-from keras.preprocessing.image import ImageDataGenerator
-from keras.preprocessing import image
+from tensorflow.keras.preprocessing.image import ImageDataGenerator
+from tensorflow.keras.preprocessing import image
 import numpy as np
 
 # Check TensorFlow version
-st.write("TensorFlow version: ", tf.__version__)
+print("TensorFlow version: ", tf.__version__)
 
 # Sidebar for file upload
 st.sidebar.title("Upload Image")
@@ -24,7 +24,7 @@ if uploaded_file is not None:
         horizontal_flip=True
     )
 
-    # Load the pre-trained model
+    # No need to load the model architecture, just load the weights
     cnn = tf.keras.models.Sequential([
         tf.keras.layers.Conv2D(filters=32, kernel_size=3, activation='relu', input_shape=[64, 64, 3]),
         tf.keras.layers.MaxPool2D(pool_size=2, strides=2),