Spaces:
Runtime error
Runtime error
fix ort model
Browse files
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 |
-
|
137 |
-
|
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 |
|