Update app.py

app.py

@@ -25,11 +25,12 @@ def translate_text(text, source_lang, target_lang):
         prompt = f"Translate the following text from {source_lang} to {target_lang}: '{text}'. Do not add any extra explanations."
         response = client.text_generation(
             prompt,
-            max_new_tokens=100,  # 불필요한 텍스트 생성을 방지하기 위해 토큰 수 제한
-            temperature=0.3,     # 창의성을 낮춰 정확한 번역 유도
-            top_p=0.9            # 높은 확률의 단어만 선택하도록 설정
+            max_new_tokens=50,  # 불필요한 텍스트 생성을 방지하기 위해 토큰 수 제한
+            temperature=0.3,    # 창의성을 낮춰 정확한 번역 유도
+            top_p=0.9           # 높은 확률의 단어만 선택하도록 설정
         )
-        return response.strip()  # 불필요한 공백 제거
+        # 번역 결과에서 불필요한 부분 제거
+        return response.strip().split("\n")[0]  # 첫 번째 줄만 반환
     except Exception as e:
         return f"오류가 발생했습니다: {str(e)}"