# MODEL_NAME="claude-3-7-sonnet-20250219"
# MODEL="bedrock/us.anthropic.claude-3-7-sonnet-20250219-v1:0"

MODEL_NAME="claude-sonnet-4-20250514"
MODEL="bedrock/us.anthropic.claude-sonnet-4-20250514-v1:0"

INPUT_PATH="<path_to_input_data_json>"
OUTPUT_PATH="<path_to_your_output>"

python inference_claude_clevr_policy.py \
  --model ${MODEL} \
  --sharegpt_dataset_path ${INPUT_PATH} \
  --output_json_path ${OUTPUT_PATH}