Update app.py

app.py

@@ -28,13 +28,17 @@ labels = [
 
 def detect_context(input_text):
     # Tokenize input text
-    inputs = tokenizer(input_text, return_tensors="np")
+    inputs = 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)
+    attention_mask = inputs["attention_mask"].astype(np.int64)
     
     # Run inference with context model
-    outputs = context_session.run(None, {"input_ids": input_ids})
+    outputs = context_session.run(None, {
+        "input_ids": input_ids
+        "attention_mask": attention_mask
+        })
     
     # Assuming the output is logits for each label
     scores = outputs[0]
@@ -91,4 +95,4 @@ gr.Interface(
     inputs="text",
     outputs="text",
     live=True
-).launch()
+).launch()