diff --git a/environment.yml b/environment.yml
index aee52f8..e6b6071 100644
--- a/environment.yml
+++ b/environment.yml
@@ -30,6 +30,7 @@ dependencies:
     - gradio==3.47.1
     - accelerate==0.20.3
     - bitsandbytes==0.37.0
+    - torchmetrics
     - scikit-image
     - visual-genome
     - wandb
diff --git a/requirements.txt b/requirements.txt
index d2676f9..063befe 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,5 +20,6 @@ sentence-transformers
 gradio==3.47.1
 accelerate==0.20.3
 bitsandbytes==0.37.0
+torchmetrics
 scikit-image
 wandb
\ No newline at end of file