Spaces:
Build error
Build error
Commit
•
bde4867
1
Parent(s):
7a8f502
Change model to ismir
Browse files- inferencemodel.py +1 -1
inferencemodel.py
CHANGED
|
@@ -21,7 +21,7 @@ from mt3 import vocabularies
|
|
| 21 |
class InferenceModel(object):
|
| 22 |
"""Wrapper of T5X model for music transcription."""
|
| 23 |
|
| 24 |
-
def __init__(self, checkpoint_path, model_type='
|
| 25 |
|
| 26 |
# Model Constants.
|
| 27 |
if model_type == 'ismir2021':
|
|
|
|
| 21 |
class InferenceModel(object):
|
| 22 |
"""Wrapper of T5X model for music transcription."""
|
| 23 |
|
| 24 |
+
def __init__(self, checkpoint_path, model_type='ismir2021'):
|
| 25 |
|
| 26 |
# Model Constants.
|
| 27 |
if model_type == 'ismir2021':
|