Update app.py

app.py

@@ -8,7 +8,7 @@ model_path = "cifar_net.pth"
 
 model = torch.load(model_path)
 
-model = MyModel()
+model = torch.nn.Module()
 model.load_state_dict(state_dict)
 model.eval()