Spaces:
Paused
Paused
Andrei-Iulian SĂCELEANU
commited on
Commit
•
5506e99
1
Parent(s):
37f6940
fix audio-mixmatch
Browse files
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(
|
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 |
|