Update app.py

app.py

@@ -11,8 +11,9 @@ translation_model_file = "./model.onnx"
 context_session = ort.InferenceSession(context_model_file)
 translation_session = ort.InferenceSession(translation_model_file)
 
-# Load tokenizer for translation model
+# 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", 
@@ -28,7 +29,7 @@ labels = [
 
 def detect_context(input_text):
     # Tokenize input text
-    inputs = tokenizer(input_text, return_tensors="np", padding=True, truncation=True, max_length=512)
+    inputs = context_tokenizer(input_text, return_tensors="np", padding=True, truncation=True, max_length=512)
     
     # Prepare input for ONNX model
     input_ids = inputs["input_ids"].astype(np.int64)