Update app.py

app.py

@@ -282,15 +282,15 @@ else:
 # Load speech-to-text model (Whisper)
 speech_to_text = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
 
-# Load Stable Diffusion model for text-to-image
-# text_to_image = StableDiffusionPipeline.from_pretrained(
-#     "runwayml/stable-diffusion-v1-5",revision="fp16"     
-# )
+#Load Stable Diffusion model for text-to-image
 text_to_image = StableDiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
-    cache_dir="./my_model_cache",  # Custom cache directory
-    revision="fp16"
+    "runwayml/stable-diffusion-v1-5",revision="fp16"     
 )
+# text_to_image = StableDiffusionPipeline.from_pretrained(
+#     "runwayml/stable-diffusion-v1-5",
+#     cache_dir="./my_model_cache",  # Custom cache directory
+#     revision="fp16"
+# )
 
 device = "cuda" if torch.cuda.is_available() else "cpu"  # This will now work since torch is imported
 text_to_image.to(device)