Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -37,7 +37,7 @@ class EndpointHandler():
|
|
37 |
"reason": "max_length was passed as something that was absolutely not a plain old int"
|
38 |
})
|
39 |
|
40 |
-
res = PROMPT_FORMAT.format(
|
41 |
return model(
|
42 |
res,
|
43 |
max_new_tokens=max_new_tokens,
|
|
|
37 |
"reason": "max_length was passed as something that was absolutely not a plain old int"
|
38 |
})
|
39 |
|
40 |
+
res = PROMPT_FORMAT.format(inputs=data['inputs'])
|
41 |
return model(
|
42 |
res,
|
43 |
max_new_tokens=max_new_tokens,
|