Spaces:

asquirous
/

tv_desktop_classifier

Runtime error

asquirous commited on Jun 21, 2023

Commit

ebab755

1 Parent(s): f64fcc8

Update if condition in app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ def classify_image(img):
     output = dict(zip(labels, map(float, probs)))
     for out in output:
         val = output[out]
-        if val*100 < 60:
             return {"Not Sure/Others": 0}
     return output

     output = dict(zip(labels, map(float, probs)))
     for out in output:
         val = output[out]
+        if val < 60:
             return {"Not Sure/Others": 0}
     return output