Update app.py

app.py

@@ -34,6 +34,16 @@ def translate_text(input_texts):
     ort_outputs = session.run(None, ort_inputs)
     output_ids = ort_outputs[0]  # Get the output token IDs
 
+    # Debug: Inspect the structure of output_ids
+    print("Output IDs shape:", output_ids.shape)
+    print("Output IDs:", output_ids)
+
+    # Ensure output_ids is in the correct format (2D array)
+    if isinstance(output_ids, list):
+        output_ids = np.array(output_ids)  # Convert list to numpy array if necessary
+    if output_ids.ndim > 2:
+        output_ids = output_ids.squeeze(0)  # Remove extra dimensions if present
+
     # Decode the output tokens
     translated_texts = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
     return translated_texts