Update app.py
Browse files
app.py
CHANGED
@@ -39,6 +39,7 @@ def init_prompt_injection_model(prompt_injection_ort_model: str, subfolder: str
|
|
39 |
prompt_injection_ort_model,
|
40 |
export=False,
|
41 |
subfolder=subfolder,
|
|
|
42 |
)
|
43 |
hf_tokenizer = AutoTokenizer.from_pretrained(prompt_injection_ort_model, subfolder=subfolder)
|
44 |
hf_tokenizer.model_input_names = ["input_ids", "attention_mask"]
|
|
|
39 |
prompt_injection_ort_model,
|
40 |
export=False,
|
41 |
subfolder=subfolder,
|
42 |
+
file_name="model.onnx"
|
43 |
)
|
44 |
hf_tokenizer = AutoTokenizer.from_pretrained(prompt_injection_ort_model, subfolder=subfolder)
|
45 |
hf_tokenizer.model_input_names = ["input_ids", "attention_mask"]
|