Update app.py

app.py

@@ -22,7 +22,21 @@ from tensorflow.keras.utils import load_img, img_to_array
 from sklearn.metrics import confusion_matrix
 
 
-model = tf.keras.models.load_model('my_model.keras')
+vgg16_model = vgg16.VGG16(include_top=False, input_shape=(224,224,3), weights='imagenet')
+vgg16_model.trainable = False
+model = Sequential()
+model.add(vgg16_model)
+model.add(Flatten())
+model.add(Dense(4096, activation='relu'))
+model.add(Dense(4096, activation='relu'))
+model.add(Dropout(0.25))
+model.add(Dense(1, activation='sigmoid'))
+
+model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['Accuracy'])
+model.load_weight("cp.ckpt")
+
+
+
 
 
 import gradio as gr