eventhorizon28 commited on
Commit
9845eee
1 Parent(s): 1d1c7b1

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -4,7 +4,7 @@ from typing import Any
4
  class EndpointHandler():
5
  def __init__(self, path=""):
6
  # create inference pipeline
7
- self.pipeline = pipeline("text-to-speech", model=path)
8
 
9
  def __call__(self, data: Any) -> Any:
10
  inputs = data.pop("inputs", data)
 
4
  class EndpointHandler():
5
  def __init__(self, path=""):
6
  # create inference pipeline
7
+ self.pipeline = pipeline("text-to-speech", model=path, device=0)
8
 
9
  def __call__(self, data: Any) -> Any:
10
  inputs = data.pop("inputs", data)