y

app.py

@@ -9,6 +9,7 @@ from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
 from two_stream_shunt_adapter import TwoStreamShuntAdapter
 from configs import T5_SHUNT_REPOS
+from PIL import Image
 
 # ─── Device & Model Setup ─────────────────────────────────────
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

requirements.txt

@@ -5,4 +5,5 @@ invisible_watermark
 torch
 transformers
 xformers
-matplotlib
+matplotlib
+pillow