Update app.py

app.py

@@ -1,21 +1,43 @@
 import os
-import torch
-import streamlit as st
-from diffusers import DiffusionPipeline
-from PIL import Image
+import subprocess
 
-# Install missing dependencies
+# Function to install packages
+def install(package):
+    subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
+
+# Install torch if not already installed
 try:
-    import diffusers
+    import torch
 except ModuleNotFoundError:
-    os.system("pip install torch torchvision diffusers transformers Pillow")
+    install("torch")
+    import torch
+
+import streamlit as st
+from diffusers import DiffusionPipeline
+from PIL import Image
 
 # Streamlit UI
 st.title("Lightweight Text-to-Image Generator (CPU Friendly)")
 st.write("Enter a description, and an AI model will generate an image!")
 
-# Load a small model (works on CPU)
+# Load the model
 @st.cache_resource()
 def load_model():
     model_id = "CompVis/txt2img-f8-large"  # Lightweight model for CPU
-    pipe = DiffusionPipeline.fr
+    pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+    pipe.to("cpu")  # Ensure the model runs on CPU
+    return pipe
+
+pipe = load_model()
+
+# User input
+text_input = st.text_input("Enter your description:")
+
+# Generate and display image
+if text_input:
+    with st.spinner("Generating image... Please wait ⏳"):
+        try:
+            image = pipe(text_input).images[0]  # Generate image
+            st.image(image, caption="Generated Image", use_column_width=True)
+        except Exception as e:
+            st.error(f"Error generating image: {e}")