Narsil HF staff commited on
Commit
af7cc76
1 Parent(s): 2681d11

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
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) -> list[int]:
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.