Spaces:
Runtime error
Runtime error
Update app/Hackathon_setup/exp_recognition_model.py
Browse files
app/Hackathon_setup/exp_recognition_model.py
CHANGED
|
@@ -19,9 +19,8 @@ classes = {0: 'ANGER', 1: 'DISGUST', 2: 'FEAR', 3: 'HAPPINESS', 4: 'NEUTRAL', 5:
|
|
| 19 |
|
| 20 |
# Example Network
|
| 21 |
class facExpRec(torch.nn.Module):
|
| 22 |
-
|
| 23 |
super().__init__()
|
| 24 |
-
|
| 25 |
self.conv1 = self.convlayer(in_channels=1, out_channels=64, kernel_size=3, max_pool=2)
|
| 26 |
self.conv2 = self.convlayer(in_channels=64, out_channels=128, kernel_size=3, max_pool=2)
|
| 27 |
self.conv3 = self.convlayer(in_channels=128, out_channels=512, kernel_size=3, max_pool=2)
|
|
|
|
| 19 |
|
| 20 |
# Example Network
|
| 21 |
class facExpRec(torch.nn.Module):
|
| 22 |
+
def __init__(self, out_features=7):
|
| 23 |
super().__init__()
|
|
|
|
| 24 |
self.conv1 = self.convlayer(in_channels=1, out_channels=64, kernel_size=3, max_pool=2)
|
| 25 |
self.conv2 = self.convlayer(in_channels=64, out_channels=128, kernel_size=3, max_pool=2)
|
| 26 |
self.conv3 = self.convlayer(in_channels=128, out_channels=512, kernel_size=3, max_pool=2)
|