Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
-from TTS.api import TTS  # Coqui TTS 라이브러리
+from TTS.api import TTS
 
 # 모델 이름
-model_name = "mradermacher/SambaLingo-Bulgarian-Base-i1-GGUF"
+model_name = "facebook/llama-7b"
 
 # 모델 및 토크나이저 로드
 try:
@@ -12,7 +12,7 @@ try:
 except Exception as e:
     raise RuntimeError(f"모델 로드 중 문제가 발생했습니다: {e}")
 
-# TTS 모델 로드 (Coqui TTS)
+# TTS 모델 로드
 try:
     tts = TTS(model_name="tts_models/bg/cv/vits", progress_bar=False)
 except Exception as e:
@@ -20,16 +20,16 @@ except Exception as e:
 
 # 텍스트 생성 및 음성 변환 함수
 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
 
-    # TTS 변환
     try:
+        # TTS 변환
         audio_path = "output.wav"
         tts.tts_to_file(text=generated_text, file_path=audio_path)
         return generated_text, audio_path
@@ -39,10 +39,10 @@ def generate_audio(input_text):
 # Gradio 인터페이스 생성
 interface = gr.Interface(
     fn=generate_audio,
-    inputs=gr.Textbox(lines=5, label="불가리아어 텍스트 입력"),
+    inputs=gr.Textbox(lines=5, label="텍스트 입력"),
     outputs=[gr.Textbox(label="생성된 텍스트"), gr.Audio(label="생성된 음성")],
-    title="불가리아어 텍스트 생성 및 음성 변환",
-    description="불가리아어 텍스트를 입력하면 텍스트를 생성하고 음성으로 변환합니다."
+    title="LLaMA 7B 텍스트 생성 및 음성 변환",
+    description="텍스트를 입력하면 생성된 텍스트를 출력하고 음성으로 변환합니다."
 )
 
 # 앱 실행