Update handler.py
Browse files- 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=
|
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
|