Update app.py

app.py

@@ -1,9 +1,4 @@
-# -*- coding: utf-8 -*-
-"""
-Created on Tue Sep 17 20:56:04 2024
 
-@author: SABARI
-"""
 # -*- coding: utf-8 -*-
 """
 Created on Tue Sep 17 19:03:17 2024
@@ -58,7 +53,8 @@ class JapaneseNER():
         sample_encoding = self.tokenizer([ 
           "鈴木は4月の陽気の良い日に、鈴をつけて熊本県の阿蘇山に登った",
           "中国では、中国共産党による一党統治が続く",
-        ], truncation=True, max_length=512, return_tensors="pt")
+        ], truncation=True,padding=True,  # Ensure all sequences are of the same length
+                                    max_length=512, return_tensors="pt")
         
         sample_encoding = {k: v.to(device) for k, v in sample_encoding.items()}
 
@@ -70,7 +66,7 @@ class JapaneseNER():
         print("Predicted labels:", predicted_label_id)
 
     def predict(self, text):
-        encoding = self.tokenizer([text], truncation=True, max_length=512, return_tensors="pt")
+        encoding = self.tokenizer([text], truncation=True,padding=True, max_length=512, return_tensors="pt")
         encoding = {k: v.to(device) for k, v in encoding.items()}
         
         # Perform prediction