Update app.py

app.py

@@ -1,50 +1,35 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
-from TTS.api import TTS
+from transformers import pipeline
 
-# 모델 이름
-model_name = "facebook/llama-7b"
+# Step 1: Initialize the text-generation pipeline
+pipe = pipeline("text-generation", model="sambanovasystems/SambaLingo-Bulgarian-Base")
 
-# 모델 및 토크나이저 로드
-try:
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-except Exception as e:
-    raise RuntimeError(f"모델 로드 중 문제가 발생했습니다: {e}")
+# Step 2: Define a function to generate Bulgarian text and convert it to TTS
+def generate_and_speak(prompt):
+    # Generate text using the model
+    generated_text = pipe(prompt, max_length=50, num_return_sequences=1)[0]["generated_text"]
+    
+    # Placeholder: Simulating TTS (you can replace this with an actual TTS library)
+    # Install a TTS library such as gTTS for better implementation
+    from gtts import gTTS
+    tts = gTTS(generated_text, lang="bg")
+    audio_file = "output.mp3"
+    tts.save(audio_file)
+    
+    return generated_text, audio_file
 
-# TTS 모델 로드
-try:
-    tts = TTS(model_name="tts_models/bg/cv/vits", progress_bar=False)
-except Exception as e:
-    raise RuntimeError(f"TTS 모델 로드 중 문제가 발생했습니다: {e}")
+# Step 3: Create Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("## Bulgarian Text Generator and TTS")
+    with gr.Row():
+        input_prompt = gr.Textbox(label="Enter a prompt in Bulgarian:")
+        with gr.Column():
+            output_text = gr.Textbox(label="Generated Text")
+            output_audio = gr.Audio(label="Generated Speech", type="file")
+    generate_button = gr.Button("Generate")
+    
+    generate_button.click(generate_and_speak, inputs=input_prompt, outputs=[output_text, output_audio])
 
-# 텍스트 생성 및 음성 변환 함수
-def generate_audio(input_text):
-    try:
-        # 텍스트 생성
-        inputs = tokenizer.encode(input_text, return_tensors="pt")
-        outputs = model.generate(inputs, max_length=100, num_return_sequences=1)
-        generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    except Exception as e:
-        return f"텍스트 생성 중 문제가 발생했습니다: {e}", None
-
-    try:
-        # TTS 변환
-        audio_path = "output.wav"
-        tts.tts_to_file(text=generated_text, file_path=audio_path)
-        return generated_text, audio_path
-    except Exception as e:
-        return f"TTS 변환 중 문제가 발생했습니다: {e}", None
-
-# Gradio 인터페이스 생성
-interface = gr.Interface(
-    fn=generate_audio,
-    inputs=gr.Textbox(lines=5, label="텍스트 입력"),
-    outputs=[gr.Textbox(label="생성된 텍스트"), gr.Audio(label="생성된 음성")],
-    title="LLaMA 7B 텍스트 생성 및 음성 변환",
-    description="텍스트를 입력하면 생성된 텍스트를 출력하고 음성으로 변환합니다."
-)
-
-# 앱 실행
+# Run the app
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()