Update app.py

app.py

@@ -27,9 +27,8 @@ hf_client = InferenceClient(
 
 @spaces.GPU
 def transcribe_summarize(audio_input, task):
-    # 입력 검증
     if audio_input is None:
-        raise gr.Error("오디오 파일이 제출되지 않았습니다! 요청을 제출하기 전에 오디오 파일을 업로드하거나 녹음해 주세요.")
+        raise gr.Error("오디오 파일이 제출되지 않았습니다!")
     
     # 음성을 텍스트로 변환
     result = pipe(
@@ -40,14 +39,36 @@ def transcribe_summarize(audio_input, task):
     )
     transcribed_text = result["text"]
     
-    # 텍스트 요약
+    # 텍스트 요약 (수정된 부분)
     try:
-        summary = hf_client.summarization(transcribed_text)
-        summary_text = summary.get("summary_text", "요약할 수 없습니다.")
+        # 요약 API 호출 파라미터 설정
+        summary_params = {
+            "text": transcribed_text,
+            "max_length": 150,
+            "min_length": 30,
+            "do_sample": False
+        }
+        
+        # API 호출
+        response = hf_client.text_generation(
+            prompt=f"Please summarize the following text:\n{transcribed_text}\n\nSummary:",
+            max_new_tokens=150,
+            temperature=0.3,
+            repetition_penalty=1.2
+        )
+        
+        summary_text = response[0]["generated_text"] if response else "요약을 생성할 수 없습니다."
+        
+        # 프롬프트 부분 제거
+        summary_text = summary_text.split("Summary:")[1].strip() if "Summary:" in summary_text else summary_text
+        
     except Exception as e:
-        summary_text = f"요약 중 오류가 발생했습니다: {str(e)}"
+        print(f"요약 생성 중 오류 발생: {str(e)}")  # 디버깅을 위한 로그
+        summary_text = f"요약 생성 중 오류가 발생했습니다: {str(e)}"
+    
+    print(f"변환된 텍스트: {transcribed_text}")  # 디버깅을 위한 로그
+    print(f"생성된 요약: {summary_text}")  # 디버깅을 위한 로그
     
-    # 결과를 리스트로 반환
     return [transcribed_text, summary_text]
 
 # CSS 스타일
@@ -67,8 +88,8 @@ file_transcribe = gr.Interface(
         ),
     ],
     outputs=[
-        gr.Textbox(label="변환된 텍스트"),
-        gr.Textbox(label="요약")
+        gr.Textbox(label="변환된 텍스트", lines=5),
+        gr.Textbox(label="요약", lines=3)
     ],
     title="받아쓰기 AI: 음성을 텍스트로 변환하고 요약하기",
     flagging_mode="never"
@@ -86,8 +107,8 @@ mic_transcribe = gr.Interface(
         ),
     ],
     outputs=[
-        gr.Textbox(label="변환된 텍스트"),
-        gr.Textbox(label="요약")
+        gr.Textbox(label="변환된 텍스트", lines=5),
+        gr.Textbox(label="요약", lines=3)
     ],
     title="받아쓰기 AI: 음성을 텍스트로 변환하고 요약하기",
     flagging_mode="never",