Update app/Hackathon_setup/exp_recognition_model.py

app/Hackathon_setup/exp_recognition_model.py

@@ -18,7 +18,8 @@ logger = logging.get_logger("transformers")
 classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5: 'SADNESS', 6: 'SURPRISE'}
 
 # Example Network
-def __init__(self, out_features=7):
+class facExpRec(torch.nn.Module):
+    def __init__(self, out_features=7):
         super().__init__()
 
         self.conv1 = self.convlayer(in_channels=1, out_channels=64, kernel_size=3)