skroed
commited on
Commit
·
245bc3c
1
Parent(s):
cb8faa1
Fix: need to provide inputs.
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -51,7 +51,7 @@ class EndpointHandler:
|
|
51 |
Returns:
|
52 |
Dict[str, str]: A dictionary with the generated audio.
|
53 |
"""
|
54 |
-
prompt = data["
|
55 |
|
56 |
generation_params = data.get("generation_params", {})
|
57 |
|
|
|
51 |
Returns:
|
52 |
Dict[str, str]: A dictionary with the generated audio.
|
53 |
"""
|
54 |
+
prompt = data["inputs"]
|
55 |
|
56 |
generation_params = data.get("generation_params", {})
|
57 |
|