asofter commited on
Commit
40d2b4b
·
verified ·
1 Parent(s): 0604028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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"]