fix

app.py

@@ -12,7 +12,7 @@ matplotlib.use('agg')
 from matplotlib import pyplot as plt
 
 resnet_18 = model.LitResnet()
-state_dict = torch.load("saved_model.pth", map_location=torch.device('cpu'))
+state_dict = torch.load("resnet.pth", map_location=torch.device('cpu'))
 resnet_18.load_state_dict(state_dict)
 resnet_18_model = resnet_18.model
 
@@ -118,7 +118,7 @@ with gr.Blocks() as demo:
             ["examples/cat.jpg", 3, True],
             ["examples/horse.jpg", 3, True],
             ["examples/airplane.jpg", 3, True],
-            ["examples/parrot.jpg", 3, True],
+            ["examples/bird.jpg", 3, True],
             ["examples/automobile.jpg", 3, True],
             ["examples/deer.jpg", 3, True],
             ["examples/frog.jpg", 3, True],