diff --git a/environment.yml b/environment.yml
index e6b6071..44774b5 100644
--- a/environment.yml
+++ b/environment.yml
@@ -30,7 +30,8 @@ dependencies:
     - gradio==3.47.1
     - accelerate==0.20.3
     - bitsandbytes==0.37.0
-    - torchmetrics
+    - pycocotools
+    - torchmetrics[detection]
     - scikit-image
     - visual-genome
     - wandb
diff --git a/requirements.txt b/requirements.txt
index 063befe..4a3376a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,6 +20,7 @@ sentence-transformers
 gradio==3.47.1
 accelerate==0.20.3
 bitsandbytes==0.37.0
-torchmetrics
+pycocotools
+torchmetrics[detection]
 scikit-image
 wandb
\ No newline at end of file