Update app.py

app.py

@@ -13,10 +13,11 @@ def generate_prompt(description: str) -> str:
 
 # 加载 ByteDance/SDXL-Lightning 模型
 pipe = DiffusionPipeline.from_pretrained("ByteDance/SDXL-Lightning")
+
 def generate_image(prompt: str):
     # 根据提示生成图像
-    prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
     image = pipe(prompt).images[0]
+    return image
 
 # 使用Whisper模型进行语音转文本
 processor = WhisperProcessor.from_pretrained("openai/whisper-large")
@@ -71,3 +72,4 @@ interface_with_audio = gr.Interface(
 
 # 使用标签页界面展示文本和语音输入
 gr.TabbedInterface([interface, interface_with_audio]).launch()
+