fix identation

model.py

@@ -24,10 +24,10 @@ def create_vggface2_model(num_classes:int=2,
     model_pred = nn.Sequential(*list(model_pred.children())[:-5])
 
     class Flatten(nn.Module):
-    def __init__(self):
+      def __init__(self):
         super(Flatten, self).__init__()
         
-    def forward(self, x):
+      def forward(self, x):
         x = x.view(x.size(0), -1)
         return x