Update app.py

app.py

@@ -130,7 +130,7 @@ method = Gradio(model_path="./checkpoint/model.pt")
 demo = gr.Interface(
     fn=method.inference,
     inputs=[
-        gr.Image(shape=(32, 32), label="Input Image", value="./samples/bird_and_plane_.jpeg"),
+        gr.Image(shape=(32, 32), label="Input Image", value="./samples/dog_cat.jpeg"),
         gr.Slider(
             minimum=0,
             maximum=1,
@@ -161,7 +161,7 @@ demo = gr.Interface(
         ),
     ],
     outputs=[
-        gr.Image(shape=(32, 32)).style(width=128, height=128),
+        gr.Image(shape=(32, 32)).style(width=256, height=256),
         gr.Label(label="Top Classes"),
     ],
     examples=[[os.path.join("./samples/", f)] for f in os.listdir("./samples/")]