Spaces:
Sleeping
Sleeping
Update classifier.py
Browse files- classifier.py +1 -1
classifier.py
CHANGED
|
@@ -14,7 +14,7 @@ def classify(tweet, event_model, hftoken, threshold):
|
|
| 14 |
|
| 15 |
|
| 16 |
# with onnx pipeline
|
| 17 |
-
onnx_classifier =
|
| 18 |
prediction = onnx_classifier(text)[0]
|
| 19 |
|
| 20 |
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
# with onnx pipeline
|
| 17 |
+
onnx_classifier = opipeline("text-classification", model=event_model, accelerator="ort")
|
| 18 |
prediction = onnx_classifier(text)[0]
|
| 19 |
|
| 20 |
|