Update app.py

app.py

@@ -90,7 +90,7 @@ def load_model(model_name: str):
         width = 768
         train_sequence_length = 16000
         weight_decay = 0.0001
-        num_labs = len(set(y_train))
+        num_labs = 38
         
         
         model_seq = nn.Sequential(
@@ -112,6 +112,9 @@ def load_model(model_name: str):
             new_head
         )
         
+        weights = torch.load('NOO_CNN_1stGEAC_m4_16kcw_best.pth')
+        joined_model.load_state_dict(weights)
+        
         return joined_model, None
 
     else: