fix code error

app.py

@@ -7,17 +7,14 @@ model = tf.keras.models.load_model('model.h5')
 
 # Define the class names
 class_names = {
-    0: 'Glioma',
-    1: 'Menin',
-    2: 'Tumor'
+    0: 'Esophagitis',
+    1: 'Dyed lifted polyps'
 }
 
 
-{'brain_glioma': 0, 'brain_menin': 1, 'brain_tumor': 2}
-
 def classify_image(image):
     # Preprocess the image
-    img_array = tf.image.resize(image, 200, 200])
+    img_array = tf.image.resize(image, [200, 200])
     img_array = tf.expand_dims(img_array, 0) / 255.0
 
     # Make a prediction