CHORE: Update requirements.txt to include torch and torchvision, and … #2

	name: CI

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

	jobs:
	lint:
	name: Lint with black
	runs-on: ubuntu-latest

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

	- name: Set up Python
	uses: actions/setup-python@v4
	with:
	python-version: '3.x'

	- name: Install black
	run: pip install black

	- name: Run black
	run: black --check .

	test:
	name: Test with pytest
	runs-on: ubuntu-latest

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

	- name: Set up Python
	uses: actions/setup-python@v4
	with:
	python-version: '3.x'

	- name: Install dependencies
	run: \|
	pip install pytest
	pip install -r requirements.txt

	- name: Run tests
	run: pytest

Provide feedback