Update app.py

app.py

@@ -67,9 +67,9 @@ class GSNet(nn.Module):
         return x
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-nclasses = 4
+nclasses = 5
 my_pretrained_model = GSNet(nclasses)
-model_path = "CI_model_new.pth"  
+model_path = "CI_model.pth"  
 my_pretrained_model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
 my_pretrained_model.eval()
 transform = transforms.Compose([