Lin0He commited on
Commit
eed5e39
·
1 Parent(s): f36ca98

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -27,7 +27,7 @@ def topk(probs, n=9):
27
  tokenId = topIx[choice][0]
28
  return int(tokenId)
29
 
30
- def model_infer(model, tokenizer, review, max_length=60):
31
  # Preprocess the init token (task designator)
32
  review_encoded = tokenizer.encode(review)
33
  result = review_encoded
 
27
  tokenId = topIx[choice][0]
28
  return int(tokenId)
29
 
30
+ def model_infer(model, tokenizer, review, max_length=300):
31
  # Preprocess the init token (task designator)
32
  review_encoded = tokenizer.encode(review)
33
  result = review_encoded