Update app.py

app.py

@@ -13,7 +13,7 @@ from datetime import datetime
 from transformers import pipeline, AutoModelForSpeechSeq2Seq, AutoProcessor
 from googlemaps import Client as GoogleMapsClient
 from gtts import gTTS
-from diffusers import StableDiffusion3Pipeline
+from diffusers import StableDiffusionPipeline
 from langchain_openai import OpenAIEmbeddings, ChatOpenAI
 from langchain_pinecone import PineconeVectorStore
 from langchain.prompts import PromptTemplate
@@ -509,16 +509,16 @@ def generate_audio_parler_tts(text):
     logging.debug(f"Audio saved to {temp_audio_path}")
     return temp_audio_path
 
-pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", torch_dtype=torch.float16)
+pipe.to(device)
 
 def generate_image(prompt):
-    image = pipe(
-        prompt,
-        negative_prompt="",
-        num_inference_steps=28,
-        guidance_scale=3.0,
-    ).images[0]
+    with torch.cuda.amp.autocast():
+        image = pipe(
+            prompt,
+            num_inference_steps=28,
+            guidance_scale=3.0,
+        ).images[0]
     return image
 
 hardcoded_prompt_1 = "Give a high quality photograph of a great looking red 2026 Bentley coupe against a skyline setting in the night, michael mann style in omaha enticing the consumer to buy this product"