chainyo commited on
Commit
6a961c3
·
1 Parent(s): 8b9d8ef

fix ort model

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -133,8 +133,8 @@ if st.session_state["init_models"]:
133
 
134
  if "ort_model" not in st.session_state:
135
  ort_model = ORTModelForSequenceClassification.from_pretrained(HUB_MODEL_PATH, from_transformers=True)
136
- # if not ONNX_MODEL_PATH.exists():
137
- # ort_model.save_pretrained(ONNX_MODEL_PATH)
138
  st.session_state["ort_model"] = ort_model
139
  st.text("✅ ONNX Model loaded.")
140
 
 
133
 
134
  if "ort_model" not in st.session_state:
135
  ort_model = ORTModelForSequenceClassification.from_pretrained(HUB_MODEL_PATH, from_transformers=True)
136
+ if not ONNX_MODEL_PATH.exists():
137
+ ort_model.save_pretrained(ONNX_MODEL_PATH)
138
  st.session_state["ort_model"] = ort_model
139
  st.text("✅ ONNX Model loaded.")
140