Update app.py

app.py

@@ -33,3 +33,15 @@ else:
 
 tokenizer = AutoTokenizer.from_pretrained(tokenizerPath)
 model = AutoModelForSequenceClassification.from_pretrained(modelPath)
+
+tokens = tokenizer.tokenize(input_text)
+token_ids = tokenizer.convert_tokens_to_ids(tokens)
+input_ids = tokenizer(input_text)
+
+batch = tokenizer(X_train, padding=True, truncation=True, max_length=512, return_tensors="pt")
+
+with torch.no_grad():
+    outputs = model(**batch)
+    predictions = TF.softmax(outputs.logits, dim=1)
+    labels = torch.argmax(predictions, dim=1)
+    labels = [model.config.id2label[label_id] for label_id in labels.tolist()]