Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ from jiwer import wer
|
|
| 30 |
|
| 31 |
@st.cache
|
| 32 |
def load_model():
|
| 33 |
-
path = 'lightning_logs/version_0/checkpoints/
|
| 34 |
onnx_model = onnx.load(path)
|
| 35 |
options = onnxruntime.SessionOptions()
|
| 36 |
options.intra_op_num_threads = 2
|
|
|
|
| 30 |
|
| 31 |
@st.cache
|
| 32 |
def load_model():
|
| 33 |
+
path = 'lightning_logs/version_0/checkpoints/frn_modified.onnx'
|
| 34 |
onnx_model = onnx.load(path)
|
| 35 |
options = onnxruntime.SessionOptions()
|
| 36 |
options.intra_op_num_threads = 2
|