diff --git a/requirements.txt b/requirements.txt
index 5bfe464..12a7609 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,5 +6,5 @@ gradio==3.41.2
 transformers==4.36.2
 nbconvert==7.8.0
 jupyterlite==0.2.1
-lightning==2.1.*
+lightning[app]==2.1.*
 jupyterlab==4.0.5