Spaces:

chainyo
/

optimum-text-classification

Runtime error

chainyo commited on Jun 6, 2022

Commit

5eceb8d

1 Parent(s): ad74093

fix ort import

Files changed (1) hide show

main.py CHANGED Viewed

@@ -7,7 +7,7 @@ Author:
 import streamlit as st
 from transformers import AutoTokenizer, AutoModel, pipeline
-from optimum.onnxruntime import ORTModelForTextClassification
 from optimum.pipelines import pipeline

 import streamlit as st
 from transformers import AutoTokenizer, AutoModel, pipeline
+from optimum.onnxruntime import ORTModelForSequenceClassification
 from optimum.pipelines import pipeline