Update autocaption main function to accept either a dataset or directory #358

	name: Test invoke-training

	on:
	push:
	branches:
	- main
	pull_request:
	workflow_dispatch:

	jobs:
	build:

	runs-on: ubuntu-latest
	strategy:
	fail-fast: false
	matrix:
	python-version: ["3.10"]

	steps:
	- uses: actions/checkout@v3
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v3
	with:
	python-version: ${{ matrix.python-version }}
	cache: pip
	cache-dependency-path: pyproject.toml
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	python -m pip install .[test]
	- name: Ruff lint
	run: \|
	ruff check --output-format=github .
	- name: Ruff format
	run: \|
	ruff format --check .
	- name: Test with pytest
	run: \|
	pytest tests --junitxml=junit/test-results-${{ matrix.python-version }}.xml -m "not cuda and not loads_model"
	- name: Upload pytest test results
	uses: actions/upload-artifact@v3
	with:
	name: pytest-results-${{ matrix.python-version }}
	path: junit/test-results-${{ matrix.python-version }}.xml
	# Use always() to always run this step to publish test results when there are test failures.
	if: ${{ always() }}

Provide feedback