Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,11 @@ import uvicorn
|
|
8 |
app = FastAPI()
|
9 |
|
10 |
# Initialize tokenizer
|
11 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
|
|
|
|
|
|
|
|
12 |
|
13 |
# Load ONNX model
|
14 |
session = InferenceSession("model.onnx")
|
|
|
8 |
app = FastAPI()
|
9 |
|
10 |
# Initialize tokenizer
|
11 |
+
tokenizer = AutoTokenizer.from_pretrained(
|
12 |
+
"Xenova/multi-qa-mpnet-base-dot-v1",
|
13 |
+
use_fast=True,
|
14 |
+
legacy=False
|
15 |
+
)
|
16 |
|
17 |
# Load ONNX model
|
18 |
session = InferenceSession("model.onnx")
|