Update app.py

app.py

@@ -90,10 +90,10 @@ class Gradio:
     def inference(
         self,
         input_img: np.array,
-        transparency: float,
-        ntop_classes: int,
-        layer_nums: List,
-        cam_for_class: str,
+        transparency: float=0.7,
+        ntop_classes: int=2,
+        layer_nums: List=[3, 4],
+        cam_for_class: str="default",
     ):
         self.model.eval()
         input_img = transform(input_img)
@@ -110,7 +110,7 @@ class Gradio:
 
         class_id = (
             prediction[0][0]
-            if cam_for_class in ["default", "", None]
+            if cam_for_class in ["default", ""]
             else classes.index(cam_for_class)
         )
         visualization = self.cam(
@@ -164,7 +164,7 @@ demo = gr.Interface(
         gr.Image(shape=(32, 32)).style(width=128, height=128),
         gr.Label(label="Top Classes"),
     ],
-    examples=[[os.path.join("samples/", f)] for f in os.listdir("samples/")]
+    examples=[[os.path.join("./samples/", f)] for f in os.listdir("samples/")]
 )