Spaces:

Ravenok
/

emotions

Running

Lord-Raven commited on Jun 7

Commit

71828e8

1 Parent(s): 8d027cb

Swapping to ONNX.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ file_name = "onnx/model_quantized.onnx"
 model = ORTModelForSequenceClassification.from_pretrained(model_id, file_name=file_name)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-onnx_classifier = pipeline(
     task="text-classification",
     model=model,
     tokenizer=tokenizer,

 model = ORTModelForSequenceClassification.from_pretrained(model_id, file_name=file_name)
 tokenizer = AutoTokenizer.from_pretrained(model_id)
+classifier = pipeline(
     task="text-classification",
     model=model,
     tokenizer=tokenizer,