Spaces:

FortuneXia
/

text-classification

Sleeping

FortuneXia commited on Jan 30

Commit

a80350e

1 Parent(s): ca9a14d

adjust

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_dir)
 model = AutoModelForSequenceClassification.from_pretrained(model_dir, config=config)
 def inference(input_text):
-    inputs = tokenizer.batch_encode_plus(
                 [input_text],
                 max_length=512,
                 truncation=True,

 model = AutoModelForSequenceClassification.from_pretrained(model_dir, config=config)
 def inference(input_text):
+    inputs = tokenizer(
                 [input_text],
                 max_length=512,
                 truncation=True,