Adding batch size argument for inference to fix memory error (#58) (#59) #313

	name: Format

	on: [push, pull_request]

	jobs:
	black:
	runs-on: ubuntu-latest

	steps:
	- name: Checkout code
	uses: actions/checkout@v4

	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: '3.12'

	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install black

	- name: Run Black
	run: \|
	black --check --diff .

Provide feedback