Spaces:

dawn17
/

cifar10

Runtime error

dawn17 commited on Aug 4, 2023

Commit

6ef1f60

1 Parent(s): e0ae716

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -110,7 +110,7 @@ class Gradio:
         class_id = (
             prediction[0][0]
-            if cam_for_class in ["default", ""]
             else classes.index(cam_for_class)
         )
         visualization = self.cam(

         class_id = (
             prediction[0][0]
+            if cam_for_class in ["default", "", None]
             else classes.index(cam_for_class)
         )
         visualization = self.cam(