zhengqilin commited on
Commit ·
96fc329
1
Parent(s): 36cfce3
handler
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -156,8 +156,8 @@ class EndpointHandler():
|
|
| 156 |
"seed": -1,
|
| 157 |
}
|
| 158 |
if data["inputs"]:
|
| 159 |
-
for field
|
| 160 |
-
if data["inputs"]
|
| 161 |
args[field] = data["inputs"][field]
|
| 162 |
# if "prompt" in data["inputs"].keys():
|
| 163 |
# prompt = data["inputs"]["prompt"]
|
|
|
|
| 156 |
"seed": -1,
|
| 157 |
}
|
| 158 |
if data["inputs"]:
|
| 159 |
+
for field in args:
|
| 160 |
+
if field in data["inputs"].keys():
|
| 161 |
args[field] = data["inputs"][field]
|
| 162 |
# if "prompt" in data["inputs"].keys():
|
| 163 |
# prompt = data["inputs"]["prompt"]
|