Frenchizer commited on
Commit
08fc7eb
·
1 Parent(s): d0ebb15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ translation_session = ort.InferenceSession(translation_model_file)
13
 
14
  # Load tokenizer for translation model and for context model
15
  tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
16
- context_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base-mnli")
17
 
18
  labels = [
19
  "general", "pharma", "legal", "technical", "UI", "user interface", "medicine",
 
13
 
14
  # Load tokenizer for translation model and for context model
15
  tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
16
+ context_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-mnli")
17
 
18
  labels = [
19
  "general", "pharma", "legal", "technical", "UI", "user interface", "medicine",