프롬프트 답변 방식 변경

app.py

@@ -61,12 +61,11 @@ async def translate(request: TranslationRequest):
         do_sample=True,
         temperature=0.7,
         top_k=50,
+        return_full_text=False,
     )
 
     # 결과물에서 필요한 부분만 추출
-    generated_text = outputs[0]["generated_text"]
-    # 'assistant\n' 다음에 오는 실제 번역 결과만 깔끔하게 잘라내기
-    translated_text = generated_text.split("assistant\n")[-1].strip()
+    translated_text = outputs[0]["generated_text"].strip()
 
     # 프론트엔드에 번역된 텍스트를 JSON 형태로 반환
     return {"translated_text": translated_text}
@@ -89,10 +88,10 @@ async def get_pronunciation(request: PronunciationRequest):
         do_sample=True,
         temperature=0.7,
         top_k=50,
+        return_full_text=False,
     )
 
-    generated_text = outputs[0]["generated_text"]
-    pronunciation_text = generated_text.split("assistant\n")[-1].strip()
+    pronunciation_text = outputs[0]["generated_text"].strip()
 
     return {"pronunciation_text": pronunciation_text}