diff --git a/requirements.txt b/requirements.txt
index d2676f9..5ce1570 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -21,4 +21,5 @@ gradio==3.47.1
 accelerate==0.20.3
 bitsandbytes==0.37.0
 scikit-image
+visual-genome
 wandb
\ No newline at end of file