3

app.py

@@ -4,6 +4,7 @@ model_0 = tf.keras.models.load_model('bestmodel.h5')
 def classify_image(inp):
     inp = inp.reshape((-1, 224, 224, 3))
     prediction = model_0.predict(inp)
+    print(prediction,"  ",prediction.argmax())
     print(prediction,"  ",prediction.argmax()," ",prediction[prediction.argmax()])
     output  = ""
     if prediction[prediction.argmax()] < 0.7: