michaellee8
/

wisper-base-jwt

Automatic Speech Recognition

hf-asr-leaderboard

Inference Endpoints

Model card Files Files and versions Community

michaellee8 commited on Jan 6, 2024

Commit

44e8c42

·

1 Parent(s): b9c964e

fix: error

Files changed (1) hide show

handler.py +3 -1

handler.py CHANGED Viewed

@@ -28,5 +28,7 @@ class EndpointHandler:
             raise RuntimeError("missing token")
         decoded = jwt.decode(token, PUBLIC_KEY, algorithms=["RS512"])
         print("received input from jti=", decoded["jti"])
-        return pipeline(*data)

             raise RuntimeError("missing token")
         decoded = jwt.decode(token, PUBLIC_KEY, algorithms=["RS512"])
         print("received input from jti=", decoded["jti"])
+        inputs = data.pop("inputs", None)
+        parameters = data.pop("parameters", {})
+        return pipeline(inputs, **parameters)