Andrei-Iulian SĂCELEANU commited on
Commit
5506e99
1 Parent(s): 37f6940

fix audio-mixmatch

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def ssl_predict2(audio_file, model_type):
108
  preds, _ = model(inputs["pixel_values"], training=False)
109
 
110
  elif model_type == "mixmatch":
111
- model = AudioMixMatch(bert_model="andrei-saceleanu/vit-base-mixmatch")
112
  model.cls_head.load_weights("./checkpoints/audio_mixmatch")
113
  preds = model(inputs["pixel_values"], training=False)
114
 
 
108
  preds, _ = model(inputs["pixel_values"], training=False)
109
 
110
  elif model_type == "mixmatch":
111
+ model = AudioMixMatch(encoder_name="andrei-saceleanu/vit-base-mixmatch")
112
  model.cls_head.load_weights("./checkpoints/audio_mixmatch")
113
  preds = model(inputs["pixel_values"], training=False)
114