Spaces:

chainyo
/

optimum-text-classification

Runtime error

chainyo commited on Jun 6, 2022

Commit

769fdce

1 Parent(s): 5eceb8d

fix ort pipeline

Files changed (1) hide show

main.py CHANGED Viewed

@@ -28,7 +28,7 @@ if "tokenizer" not in st.session_state:
     st.session_state["tokenizer"] = tokenizer
 if "ort_model" not in st.session_state:
-    ort_model = ORTModelForTextClassification.from_pretrained(MODEL_PATH, from_transformers=True)
     st.session_state["ort_model"] = ort_model
 if "pt_model" not in st.session_state:

     st.session_state["tokenizer"] = tokenizer
 if "ort_model" not in st.session_state:
+    ort_model = ORTModelForSequenceClassification.from_pretrained(MODEL_PATH, from_transformers=True)
     st.session_state["ort_model"] = ort_model
 if "pt_model" not in st.session_state: