zzarif commited on
Commit
efe5551
·
1 Parent(s): 58c091e

new model test

Browse files
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer
4
  import numpy as np
5
 
6
  # Load the ONNX model
7
- onnx_model_path = "fine-tuned_all-mpnet-base-v2_quantized.onnx"
8
  ort_session = ort.InferenceSession(onnx_model_path)
9
 
10
  # Load the tokenizer
 
4
  import numpy as np
5
 
6
  # Load the ONNX model
7
+ onnx_model_path = "fine-tuned_all-distilroberta-v1_quantized.onnx"
8
  ort_session = ort.InferenceSession(onnx_model_path)
9
 
10
  # Load the tokenizer
fine-tuned_all-mpnet-base-v2_quantized.onnx → fine-tuned_all-distilroberta-v1_quantized.onnx RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0ef30cf93af3180de7f8074c4a698e4dd48bb5dee2382e5aad2fb04311ab5633
3
- size 110266078
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1302f752e0972c9dd6db3658176bb1932d3edbc0961f4d3bf5f0143ebd468ca
3
+ size 82521329