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 |
|