ensure env installed

app.py

@@ -6,14 +6,22 @@ from pathlib import Path
 from PIL import Image
 import numpy as np
 
-def ensure_transformers_installed():
+def ensure_env_installed():
     try:
         import transformers
+        import torchvision
+        import diffusers
+        import einops
     except ImportError:
-        import subprocess, sys
-        subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers==4.54.0"])
-
-ensure_transformers_installed()
+        import subprocess
+        import sys
+        subprocess.check_call([sys.executable, "-m", "pip", "install", 
+                               "transformers==4.54.0",
+                               "torchvision==0.22.1",
+                               "diffusers==0.34.0",
+                               "einops==0.8.1"])
+
+ensure_env_installed()
 
 # Global model variable
 model_zoo = {

requirements.txt

@@ -10,7 +10,4 @@ torch==2.7.1
 torchvision==0.22.1
 transformers==4.54.0
 diffusers==0.34.0
-einops==0.8.1
-pillow==11.3.0
-numpy==2.2.6
-accelerate==1.9.0
+einops==0.8.1