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