Update app.py

app.py

@@ -132,7 +132,7 @@ class ResNet18(nn.Module):
         x = self.fc(x)
         return x
 
-model = ResNet18(1, 7)
+model = ResNet18(3, 7)
 
 checkpoint = torch.load('ham1.ckpt', map_location=torch.device('cpu'))