Update model.py

model.py

@@ -3,7 +3,7 @@ import torch.nn as nn
 from huggingface_hub import hf_hub_download
 
 class SimpleCNN(nn.Module):
-    def __init__(self, model_type='f', num_classes=4):
+    def __init__(self, model_type='f', num_classes=6):
         super(SimpleCNN, self).__init__()
         self.num_classes = num_classes
         self.model_type = model_type