Update model.py
Browse files
model.py
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
This is just an example of what people would submit for inference.
|
3 |
"""
|
4 |
import os
|
5 |
-
from typing import Dict
|
6 |
|
7 |
import torch
|
8 |
from s3prl.downstream.runner import Runner
|
@@ -19,7 +19,7 @@ class PreTrainedModel(Runner):
|
|
19 |
ckp["Args"].device = "cpu" # Just to try in my computer
|
20 |
Runner.__init__(self, ckp["Args"], ckp["Config"])
|
21 |
|
22 |
-
def __call__(self, inputs) ->
|
23 |
"""
|
24 |
Args: inputs (:obj:`np.array`): The raw waveform of audio received. By
|
25 |
default at 16KHz.
|
|
|
2 |
This is just an example of what people would submit for inference.
|
3 |
"""
|
4 |
import os
|
5 |
+
from typing import Dict, List
|
6 |
|
7 |
import torch
|
8 |
from s3prl.downstream.runner import Runner
|
|
|
19 |
ckp["Args"].device = "cpu" # Just to try in my computer
|
20 |
Runner.__init__(self, ckp["Args"], ckp["Config"])
|
21 |
|
22 |
+
def __call__(self, inputs) -> List[int]:
|
23 |
"""
|
24 |
Args: inputs (:obj:`np.array`): The raw waveform of audio received. By
|
25 |
default at 16KHz.
|