Hugging Face's logo

Spaces:
jiang20
/
test
Runtime error

App Files Community
jiang20 commited on
Commit
0fc4827
·
1 Parent(s): 90a0539

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def print_bn():
27
 
28
  model = BadNet(10)
29
  model.load_state_dict(torch.load('./cifar10_clean.pth'))
30
- # model_type = 0
31
  model.eval()
32
 
33
  transform_nor = transforms.Compose([transforms.ToTensor(), transforms.Resize((32,32)), transforms.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261))])
 
27
 
28
  model = BadNet(10)
29
  model.load_state_dict(torch.load('./cifar10_clean.pth'))
30
+ model_type = 0
31
  model.eval()
32
 
33
  transform_nor = transforms.Compose([transforms.ToTensor(), transforms.Resize((32,32)), transforms.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261))])