Update app.py

app.py

@@ -25,7 +25,7 @@ def print_bn():
 
 model = BadNet(10)
 model.load_state_dict(torch.load('./cifar10_clean.pth'))
-# model_type = 0
+model_type = 0
 model.eval()
 
 transform_nor = transforms.Compose([transforms.ToTensor(), transforms.Resize((32,32)), transforms.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261))])