Update app.py

app.py

@@ -23,6 +23,10 @@ def gradio_predict(input_text):
     input_ids = np.array(tokenized_input["input_ids"], dtype=np.int64)  # Shape should be [1, 512]
     attention_mask = np.array(tokenized_input["attention_mask"], dtype=np.int64)  # Shape should be [1, 512]
 
+    # Ensure input is in the correct shape and format for ONNX
+    input_ids = input_ids.astype(np.int64)
+    attention_mask = attention_mask.astype(np.int64)
+    
     # Prepare decoder input ids if required by your model
     decoder_input_ids = input_ids  # Adjust as needed based on model requirements