File size: 835 Bytes
5210503
 
 
 
 
 
fa05e9e
73c9e39
5210503
 
 
 
 
 
 
 
 
 
 
 
73c9e39
 
5210503
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from speechbrain.pretrained import EncoderClassifier
from typing import Dict, List, Any


class EndpointHandler:
    def __init__(self, path=""):
        self.model = EncoderClassifier.from_hparams("speechbrain/lang-id-voxlingua107-ecapa")
        print('model loaded')

    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
        """
         data args:
              inputs (:obj: `str`)
              date (:obj: `str`)
        Return:
              A :obj:`list` | `dict`: will be serialized and returned
        """
        # get inputs
        audio_url = data.pop("audio_url", data)

        print('audio_url', audio_url)

        # check if date exists and if it is a holiday
        # run normal prediction
        output = self.model.classify_file(audio_url)
        return {"prediction": output[1].exp()}