Update app.py

app.py

@@ -18,7 +18,7 @@ load_dotenv()
 # Constants
 HF_TOKEN = os.getenv("HF_TOKEN")
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-IMAGE_GENERATION_SPACE_NAME = "stabilityai/stable-diffusion-3.5-large-turbo"
+IMAGE_GENERATION_SPACE_NAME = "stabilityai/stable-diffusion-3.5-large-turbo"  # Updated to correct Space
 SUPPORTED_FORMATS = ["mp3", "wav", "ogg", "flac", "aac", "m4a"]
 
 # Pydantic Models
@@ -124,9 +124,9 @@ def generate_audio(text, language_code, speaker, path='test_audio.mp3'):
 
 def generate_image(prompt, path='test_image.png'):
     try:
-        client = Client(IMAGE_GENERATION_SPACE_NAME, hf_token=HF_TOKEN)
+        client = Client("stabilityai/stable-diffusion-3.5-large-turbo", hf_token=HF_TOKEN)
         result = client.predict(
-            prompt=prompt,
+            prompt=prompt,  # Text prompt for image generation
             width=1280,
             height=720,
             api_name="/generate_image"