Update app.py

app.py

@@ -4,15 +4,21 @@ import threading
 import base64
 from io import BytesIO
 from groq import Groq
+from diffusers import StableDiffusionPipeline
+import torch
 
 # 🔹 Initialize Groq API Client (FREE)
 groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-# 🔹 Load Text-to-Image Models (Restoring Multi-Image Generation)
+# 🔹 Load Text-to-Image Models (Updated to use diffusers for Stable Diffusion)
 model1 = gr.load("models/prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA")
 model2 = gr.load("models/Purz/face-projection")
-model3 = gr.load("models/stablediffusion/stable-diffusion-2-1")  # ✅ Fixed model
 
+# ✅ Load Stable Diffusion Model Properly (Using diffusers)
+model3 = StableDiffusionPipeline.from_pretrained(
+    "stablediffusion/stable-diffusion-2-1",
+    torch_dtype=torch.float16
+).to("cuda")  # Move to GPU if available
 
 # 🔹 Stop Event for Threading
 stop_event = threading.Event()
@@ -52,7 +58,10 @@ def generate_images(prompt):
     stop_event.clear()
     img1 = model1.predict(prompt)
     img2 = model2.predict(prompt)
-    img3 = model3.predict(prompt)
+
+    # ✅ Fix: Use Stable Diffusion correctly
+    img3 = model3(prompt).images[0]  # Get first image
+
     return img1, img2, img3
 
 # 🔹 Clear All Fields