Update app.py

app.py

@@ -1,18 +1,21 @@
 import gradio as gr
-from transformers import pipeline
+from gtts import gTTS
+from tempfile import NamedTemporaryFile
 
-# 모델을 pipeline으로 초기화합니다.
-text_gen = pipeline("text-generation", model="stabilityai/sdxl-turbo")
-
-def generate_text(prompt):
-    # 생성된 텍스트를 리턴합니다.
-    return text_gen(prompt, max_length=50, do_sample=True)[0]['generated_text']
+def text_to_speech(text, language):
+    tts = gTTS(text, lang=language)
+    with NamedTemporaryFile(suffix=".mp3", delete=False) as audio_file:
+        tts.save(audio_file.name)
+        return audio_file.name
 
 # Gradio 인터페이스를 생성합니다.
 iface = gr.Interface(
-    fn=generate_text,
-    inputs=gr.Textbox(lines=2, placeholder="Enter a prompt to generate text"),
-    outputs=gr.Textbox(),
+    fn=text_to_speech, 
+    inputs=[
+        gr.Textbox(lines=2, placeholder="Enter text for speech generation"), 
+        gr.Dropdown(label="Language", choices=["en", "es", "de", "fr"])
+    ], 
+    outputs=gr.Audio(type="file", label="Generated Speech"),
 )
 
 # 인터페이스를 실행합니다.