Update app.py

app.py

@@ -18,7 +18,11 @@ def extract_text_from_pdf(file_path):
 
 def predict_class(text):
     try:
-        inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
+        # Truncate text to maximum length of 4096 tokens
+        max_length = 4096
+        truncated_text = text[:max_length]
+
+        inputs = tokenizer(truncated_text, return_tensors="pt", padding=True, truncation=True, max_length=max_length)
         with torch.no_grad():
             outputs = model(**inputs)
             logits = outputs.logits