MrOvkill commited on
Commit
f72c5f5
1 Parent(s): 3c57069

Update handler.py

Browse files
Files changed (1) hide show
  1. 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(do_sample=False)
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,