Max Length

README.md

@@ -1,6 +1,6 @@
 ---
-title: Cd Pi
-emoji: 😻
+title: Vera
+emoji: 🏔️
 colorFrom: green
 colorTo: blue
 sdk: gradio

app.py

@@ -57,7 +57,7 @@ class Interactive:
                 'score': 0.5,
                 'score_calibrated': 0.5,
             }
-        input_ids = self.tokenizer.batch_encode_plus([statement], return_tensors='pt', padding='longest').input_ids.to(device)
+        input_ids = self.tokenizer.batch_encode_plus([statement], return_tensors='pt', padding='longest', truncation='longest_first', max_length=128).input_ids.to(device)
         with torch.no_grad():
             output = self.model(input_ids)
             last_hidden_state = output.last_hidden_state.to(device) # (B=1, L, D)