Update requirements.txt

requirements.txt

@@ -1,36 +1,37 @@
-# Core dependencies
-numpy==1.24.3
-websockets==11.0.3
-aiohttp==3.8.5
-aiosignal==1.3.1
-async-timeout==4.0.3
-attrs==23.1.0
-charset-normalizer==3.2.0
-frozenlist==1.4.0
-multidict==6.0.4
-fastapi==0.100.1
-uvicorn==0.23.2
-jinja2==3.1.2
-humanize==4.7.0
-yarl==1.9.2
-python-multipart
-# AI/ML dependencies
-torch==2.0.1
-transformers==4.31.0
-scipy==1.11.2
-
-# Runtime dependencies
-pathlib==1.0.1
-psutil==5.9.5  # For system monitoring
-ujson==5.8.0   # Fast JSON processing
-tqdm==4.65.0   # Progress bars
-colorama==0.4.6  # Terminal colors
-accelerate
-# Development dependencies (not needed in production)
-mypy==1.5.1
-pytest==7.4.0
-black==23.7.0
-pylint==2.17.5
-uvicorn
-fastapi
+# Core dependencies
+numpy==1.24.3
+websockets==11.0.3
+aiohttp==3.8.5
+aiosignal==1.3.1
+async-timeout==4.0.3
+attrs==23.1.0
+charset-normalizer==3.2.0
+frozenlist==1.4.0
+multidict==6.0.4
+fastapi==0.100.1
+uvicorn==0.23.2
+jinja2==3.1.2
+humanize==4.7.0
+yarl==1.9.2
+python-multipart
+# AI/ML dependencies
+torch==2.0.1
+transformers==4.31.0
+scipy==1.11.2
+einops
+timm
+# Runtime dependencies
+pathlib==1.0.1
+psutil==5.9.5  # For system monitoring
+ujson==5.8.0   # Fast JSON processing
+tqdm==4.65.0   # Progress bars
+colorama==0.4.6  # Terminal colors
+accelerate
+# Development dependencies (not needed in production)
+mypy==1.5.1
+pytest==7.4.0
+black==23.7.0
+pylint==2.17.5
+uvicorn
+fastapi
 pillow