Update requirements.txt

requirements.txt

@@ -10,6 +10,7 @@ contourpy==1.2.1
 cycler==0.12.1
 dnspython==2.6.1
 email_validator==2.2.0
+fastai==2.7.1
 fastapi==0.111.0
 fastapi-cli==0.0.4
 ffmpy==0.3.2
@@ -38,6 +39,7 @@ orjson==3.10.5
 packaging==24.1
 pandas==2.2.2
 pillow==10.3.0
+pip==24.1.2
 pydantic==2.7.4
 pydantic_core==2.18.4
 pydub==0.25.1
@@ -60,7 +62,9 @@ six==1.16.0
 sniffio==1.3.1
 starlette==0.37.2
 tomlkit==0.12.0
-toolz==0.12.1
+toolz==0.12.
+torch>=1.7.0,<1.12
+torchvision>=0.8.2,<0.13
 tqdm==4.66.4
 typer==0.12.3
 typing_extensions==4.12.2
@@ -70,6 +74,4 @@ urllib3==2.2.2
 uvicorn==0.30.1
 uvloop==0.19.0
 watchfiles==0.22.0
-websockets==11.0.3
-torch==2.2.0
-fastai==2.7.1
+websockets==11.0.3