Frenchizer commited on
Commit
6754320
·
verified ·
1 Parent(s): 4c4d9f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ context_session = ort.InferenceSession(context_model_file)
12
  translation_session = ort.InferenceSession(translation_model_file)
13
 
14
  # Load tokenizers for context and translation models
15
- context_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base-mnli")
16
  translation_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
17
 
18
  labels = [
 
12
  translation_session = ort.InferenceSession(translation_model_file)
13
 
14
  # Load tokenizers for context and translation models
15
+ context_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-mnli")
16
  translation_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
17
 
18
  labels = [