diff --git a/requirements.txt b/requirements.txt
index 2404caa..0cd8b07 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,4 +13,7 @@ transformers==4.19.2
 webdataset==0.2.5
 open-clip-torch==2.7.0
 gradio==3.11
+kornia==0.6
+invisible-watermark>=0.1.5
+streamlit-drawable-canvas==0.8.0
 -e .