Update app.py

app.py

@@ -31,11 +31,10 @@ else:
 def classify_text(text):
     try:
         # Tokenize the input text
-        inputs = tokenizer(text, return_tensors="pt")
+        inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
         # Pass the inputs to the model
-        outputs = model(**inputs)
-        # Get the logits and probabilities
-        logits = outputs.logits
+        logits = model(**inputs)
+        # Get the probabilities
         probabilities = torch.softmax(logits, dim=-1).tolist()[0]
         # Get the predicted class
         predicted_class = torch.argmax(logits, dim=-1).item()