Update app.py

app.py

@@ -1,17 +1,17 @@
 import os
-import subprocess
 
-# Function to install packages
-def install(package):
-    subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
-
-# Install torch if not already installed
+# Install missing dependencies
 try:
     import torch
-except ModuleNotFoundError:
-    install("torch")
-    import torch
-
+    import torchvision
+    import diffusers
+    import transformers
+    from PIL import Image
+except ModuleNotFoundError as e:
+    missing_module = str(e).split("'")[1]
+    os.system(f"pip install {missing_module}")
+
+import torch
 import streamlit as st
 from diffusers import DiffusionPipeline
 from PIL import Image