Spaces:

ADobrovsky
/

Plant_Disease_Classification_Project

Runtime error

ADobrovsky commited on Jan 27, 2023

Commit

9b9738e

•

1 Parent(s): 5a7f02e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ STD = [0.229, 0.224, 0.225]
 from torchvision.models import resnext101_64x4d
 model = resnext101_64x4d()
-model.fc = nn.Linear(model.fc.in_features, 88)
 if (torch.cuda.is_available()):
 	model.load_state_dict(torch.load(MODEL_NAME+'-model-1.pt'))

 from torchvision.models import resnext101_64x4d
 model = resnext101_64x4d()
+model.fc = torch.nn.Linear(model.fc.in_features, 88)
 if (torch.cuda.is_available()):
 	model.load_state_dict(torch.load(MODEL_NAME+'-model-1.pt'))