Spaces:

SIB
/

violence_api

Sleeping

ZELEFACK commited on Aug 2, 2022

Commit

c45ab5a

1 Parent(s): 38b165d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ def classify_image(inp):
     inp = inp.reshape((-1, 224, 224, 3))
     prediction = model_0.predict(inp)
     output  = ""
-    if prediction[0][prediction.argmax()] < 0.8:
       output = "bonne image"
     elif prediction.argmax() == 0:
       output = "Rifle violence"

     inp = inp.reshape((-1, 224, 224, 3))
     prediction = model_0.predict(inp)
     output  = ""
+    if prediction[0][prediction.argmax()] < 0.84:
       output = "bonne image"
     elif prediction.argmax() == 0:
       output = "Rifle violence"