diff --git a/requirements.txt b/requirements.txt
index eff6220..3474393 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,10 +1,6 @@
 huggingface_hub
-torch
-torchvision
-torchaudio
-ultralytics
+ ultralytics
 albumentations
-
 opencv-python
 matplotlib
 PyYAML
@@ -17,4 +13,5 @@ msgpack
 rstream
 onnxruntime-gpu
 netron
-pycuda
\ No newline at end of file
+pycuda
+tensorrt
\ No newline at end of file