Handle different model output sizes

app.py

@@ -136,7 +136,12 @@ def process_func(x: np.ndarray, sampling_rate: int) -> dict:
         # run through model
         with torch.no_grad():
             y = model(y)
-            y = torch.hstack([y[1], y[2]])
+            if len(y) == 2:
+                # Age-gender model
+                y = torch.hstack([y[1], y[2]])
+            else:
+                # Expression model
+                y = y[1]
 
         # convert to numpy
         y = y.detach().cpu().numpy()