Update model_helper.py
Browse files- model_helper.py +3 -3
model_helper.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
# @title Model helper
|
2 |
-
import spaces # for zero-GPU
|
3 |
|
4 |
import os
|
5 |
from collections import Counter
|
@@ -21,7 +21,7 @@ from model.ymt3 import YourMT3
|
|
21 |
|
22 |
|
23 |
|
24 |
-
|
25 |
def load_model_checkpoint(args=None):
|
26 |
parser = argparse.ArgumentParser(description="YourMT3")
|
27 |
# General
|
@@ -122,7 +122,7 @@ def load_model_checkpoint(args=None):
|
|
122 |
model.load_state_dict(new_state_dict, strict=False)
|
123 |
return model.eval()
|
124 |
|
125 |
-
|
126 |
def transcribe(model, audio_info):
|
127 |
t = Timer()
|
128 |
|
|
|
1 |
# @title Model helper
|
2 |
+
# import spaces # for zero-GPU
|
3 |
|
4 |
import os
|
5 |
from collections import Counter
|
|
|
21 |
|
22 |
|
23 |
|
24 |
+
|
25 |
def load_model_checkpoint(args=None):
|
26 |
parser = argparse.ArgumentParser(description="YourMT3")
|
27 |
# General
|
|
|
122 |
model.load_state_dict(new_state_dict, strict=False)
|
123 |
return model.eval()
|
124 |
|
125 |
+
|
126 |
def transcribe(model, audio_info):
|
127 |
t = Timer()
|
128 |
|