diff --git a/requirements.txt b/requirements.txt
index bc64475..e195b53 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,8 +1,9 @@
 git+https://github.com/huggingface/diffusers.git@d1c575ad7ee0390c2735f50cc59a79aae666567a#egg=diffusers
-SwissArmyTransformer
+SwissArmyTransformer==0.4.11 # Inference
 torch==2.4.0
 torchvision==0.19.0
-streamlit==1.37.0
+gradio==4.40.0 # For HF gradio demo
+streamlit==1.37.0 # For web demo
 opencv-python==4.10
 imageio-ffmpeg==0.5.1
 openai==1.38.0