Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -32,7 +32,7 @@ class EndpointHandler():
|
|
32 |
})
|
33 |
|
34 |
res = PROMPT_FORMAT.format(do_sample=False)
|
35 |
-
|
36 |
res,
|
37 |
max_new_tokens=max_new_tokens,
|
38 |
do_sample=False
|
|
|
32 |
})
|
33 |
|
34 |
res = PROMPT_FORMAT.format(do_sample=False)
|
35 |
+
return model(
|
36 |
res,
|
37 |
max_new_tokens=max_new_tokens,
|
38 |
do_sample=False
|