Thanh-Lam commited on
Commit
4b131eb
·
1 Parent(s): 3c8c19b

Fix: use correct parameter names num_genders and num_dialects

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,8 +76,8 @@ class MultiModelProfiler:
76
 
77
  model = MultiTaskSpeakerModel(
78
  model_name=encoder_name,
79
- num_gender_classes=2,
80
- num_dialect_classes=3,
81
  dropout=0.1,
82
  freeze_encoder=True
83
  )
 
76
 
77
  model = MultiTaskSpeakerModel(
78
  model_name=encoder_name,
79
+ num_genders=2,
80
+ num_dialects=3,
81
  dropout=0.1,
82
  freeze_encoder=True
83
  )