Spaces:
Sleeping
Sleeping
Update infer/utils_infer.py
Browse files- infer/utils_infer.py +1 -1
infer/utils_infer.py
CHANGED
|
@@ -223,7 +223,7 @@ def load_model(
|
|
| 223 |
device=device,
|
| 224 |
):
|
| 225 |
if vocab_file == "":
|
| 226 |
-
vocab_file =
|
| 227 |
tokenizer = "custom"
|
| 228 |
|
| 229 |
print("\nvocab : ", vocab_file)
|
|
|
|
| 223 |
device=device,
|
| 224 |
):
|
| 225 |
if vocab_file == "":
|
| 226 |
+
vocab_file = "infer/examples/vocab.txt"
|
| 227 |
tokenizer = "custom"
|
| 228 |
|
| 229 |
print("\nvocab : ", vocab_file)
|