jayebaku commited on
Commit
4354e0b
·
verified ·
1 Parent(s): 675b04e

Update classifier.py

Browse files
Files changed (1) hide show
  1. classifier.py +1 -1
classifier.py CHANGED
@@ -16,7 +16,7 @@ def classify(tweet, event_model, hftoken, threshold):
16
  # with onnx pipeline
17
  onnx_classifier = opipeline("text-classification", model=event_model, accelerator="ort",
18
  batch_size=512, token=hftoken, device="cpu")
19
- prediction = onnx_classifier(text)[0]
20
 
21
 
22
  results["text"] = tweet
 
16
  # with onnx pipeline
17
  onnx_classifier = opipeline("text-classification", model=event_model, accelerator="ort",
18
  batch_size=512, token=hftoken, device="cpu")
19
+ prediction = onnx_classifier(tweet)[0]
20
 
21
 
22
  results["text"] = tweet