Spaces:

Frenchizer
/

space_2

Build error

Frenchizer commited on Jan 18, 2025

Commit

08fc7eb

1 Parent(s): d0ebb15

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ translation_session = ort.InferenceSession(translation_model_file)
 # Load tokenizer for translation model and for context model
 tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
-context_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base-mnli")
 labels = [
     "general", "pharma", "legal", "technical", "UI", "user interface", "medicine",

 # Load tokenizer for translation model and for context model
 tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
+context_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-mnli")
 labels = [
     "general", "pharma", "legal", "technical", "UI", "user interface", "medicine",