Spaces:

D3V1L1810
/

ART_Detection

Runtime error

D3V1L1810 commited on Sep 3, 2024

Commit

2e1b199

verified ·

1 Parent(s): 92b443a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def detect_objects(images):
                         top1_class = res.probs.top1  # Predicted class
                         top1_confidence = res.probs.top1conf
                         print(top1_class)
-                        name.append([classes[top1_class]])
         else:
             names.append(['None'])
         # print(names)

                         top1_class = res.probs.top1  # Predicted class
                         top1_confidence = res.probs.top1conf
                         print(top1_class)
+                        names.append([classes[top1_class]])
         else:
             names.append(['None'])
         # print(names)