H-H-E commited on
Commit
f067669
1 Parent(s): 072ee69

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +11 -11
handler.py CHANGED
@@ -1,22 +1,22 @@
1
  from typing import Dict, List, Any
2
  from bark import SAMPLE_RATE, generate_audio, preload_models
3
 
4
- class EndpointHandler():
 
5
  def __init__(self, path=""):
6
- preload_models()
7
 
8
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
9
  """
10
- data args:
11
- inputs (:obj: `str`)
12
- date (:obj: `str`)
13
- Return:
14
- A :obj:`list` | `dict`: will be serialized and returned
15
  """
16
  # get inputs
17
- text_prompt = data.pop("inputs",data)
18
 
19
- # run normal prediction
20
- speech_array = generate_audio(text_prompt)
21
  return speech_array
22
-
 
1
  from typing import Dict, List, Any
2
  from bark import SAMPLE_RATE, generate_audio, preload_models
3
 
4
+
5
+ class EndpointHandler:
6
  def __init__(self, path=""):
7
+ preload_models()
8
 
9
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
10
  """
11
+ data args:
12
+ inputs (:obj: `str`)
13
+ date (:obj: `str`)
14
+ Return:
15
+ A :obj:`list` | `dict`: will be serialized and returned
16
  """
17
  # get inputs
18
+ text_prompt = data.pop("inputs", data)
19
 
20
+ # run normal prediction
21
+ speech_array = generate_audio(text_prompt)
22
  return speech_array