tv_desktop_classifier_duplicate

Runtime error

asquirous commited on Jun 21, 2023

Commit

ef69052

1 Parent(s): 25751a2

Fix probability of "Not Sure/Others"

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ def classify_image(img):
         if val*100 < 60:
             count += 1
         if count == 2:
-            return {"Not sure/Others": 100}
     return output
 image = gr.inputs.Image(shape=(224, 224))

         if val*100 < 60:
             count += 1
         if count == 2:
+            return {"Not sure/Others": 0}
     return output
 image = gr.inputs.Image(shape=(224, 224))