From fd324283e5ef2e7f1efce87d5ace6b3c6d63df61 Mon Sep 17 00:00:00 2001
From: Nguyen Thuan Duong <56673172+ThuanNaN@users.noreply.github.com>
Date: Mon, 20 Jan 2025 04:30:12 +0700
Subject: [PATCH] Update env

---
 environment.yml  | 1 +
 requirements.txt | 1 +
 2 files changed, 2 insertions(+)

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