Update app.py

app.py

@@ -24,7 +24,7 @@
 # from huggingface_hub import login
 
 # # Check if the token is already set in the environment variables
-# _hftoken = os.getenv("HF_TOKEN")
+# hf_token = os.getenv("HF_TOKEN")
 
 # if hf_token is None:
 #     # If the token is not set, prompt for it (this should be done securely)
@@ -888,12 +888,14 @@ from datetime import datetime
 import numpy as np
 from gtts import gTTS
 from googlemaps import Client as GoogleMapsClient
-from diffusers import StableDiffusion3Pipeline
+from diffusers import StableDiffusionPipeline
 import concurrent.futures
 from PIL import Image
+from huggingface_hub import login
 
 # Check if the token is already set in the environment variables
 hf_token = os.getenv("HF_TOKEN")
+
 if hf_token is None:
     # If the token is not set, prompt for it (this should be done securely)
     print("Please set your Hugging Face token in the environment variables.")
@@ -1302,7 +1304,7 @@ def generate_audio_elevenlabs(text):
         return None
 
 # Stable Diffusion setup
-pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-base", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 
 def generate_image(prompt):
@@ -1368,3 +1370,4 @@ demo.launch(share=True)
 
 
 
+