Spaces:

iimran
/

Abuse-Detection

Running

iimran commited on Mar 22

Commit

7e3a83e

verified ·

1 Parent(s): 13ca2f2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ class ONNXInferencePipeline:
         # Initialize tokenizer
         self.tokenizer = Tokenizer.from_file(self.tokenizer_path)
-        self.max_len = self.config["tokenizer"]["max_len"]
         # Initialize ONNX runtime session
         self.session = ort.InferenceSession(self.onnx_path)

         # Initialize tokenizer
         self.tokenizer = Tokenizer.from_file(self.tokenizer_path)
+        self.max_len = 256
         # Initialize ONNX runtime session
         self.session = ort.InferenceSession(self.onnx_path)