Update app.py

app.py

@@ -23,7 +23,7 @@ model.compile(optimizer=tf.keras.optimizers.Adam(),
              loss=tf.keras.losses.CategoricalCrossentropy(),
              metrics=[tf.keras.metrics.MeanSquaredError(), tf.keras.metrics.AUC(), tf.keras.metrics.CategoricalAccuracy()])
 
-model = tf.keras.models.load_model('my_model.h5', compile=False)
+model = tf.keras.models.load_model('my_model_1.h5', compile=False)
 
 
 def classify_image(image):
@@ -35,8 +35,6 @@ def classify_image(image):
     image_tensor = image_tensor / 255.0
     prediction = model.predict(image_tensor)
     print(prediction)
-    print(prediction.argmax)
-    
     prediction_label = str(prediction.argmax())
     return prediction_label