Update app.py

app.py

@@ -12,9 +12,8 @@ classes = ['Colon Adenocarcinoma', 'Colon Benign Tissue', 'Lung Adenocarcinoma',
 # Function to preprocess the uploaded image and make predictions
 def predict(img):
     img = img.resize((224, 224))
-    img_array = image.img_to_array(img)
-    img_array = np.expand_dims(img_array, axis=0)
-    img_array = img_array / 255.0  # Normalize the image
+    img_array = tf.keras.preprocessing.image.img_to_array(img)
+    img_array = tf.expand_dims(img_array, 0)
 
     predictions = model.predict(img_array)
     predicted_class = classes[np.argmax(predictions[0])]