4 essais

app.py

@@ -4,10 +4,8 @@ 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:
+    if prediction[0][prediction.argmax()] < 0.7:
       output = "bonne image"
     elif prediction.argmax() == 0:
       output = "Rifle violence"