Update classifier.py

classifier.py

@@ -14,7 +14,8 @@ def classify(tweet, event_model, hftoken, threshold):
 
 
     # with onnx pipeline 
-    onnx_classifier = opipeline("text-classification", model=event_model, accelerator="ort")
+    onnx_classifier = opipeline("text-classification", model=event_model, accelerator="ort", 
+                                batch_size=512, token=hftoken, device="cpu")
     prediction = onnx_classifier(text)[0]