add

app.py

@@ -45,10 +45,20 @@ def main():
     user_input = st.chat_input("메시지를 입력하세요...")
     
     if user_input:
-        messages = [{"role": "user", "content": user_input}]
-        response = pipe(messages)
+        # 사용자 메시지 표시
+        with st.chat_message("user"):
+            st.write(user_input)
+
+        # 프롬프트 생성
+        prompt = f"Human: {user_input}\n\nAssistant:"
+        
+        # 모델 생성
+        response = pipe(prompt, max_length=2048, temperature=0.7, do_sample=True)
         bot_text = response[0]["generated_text"]
         
+        # Assistant 응답만 추출
+        bot_text = bot_text.split("Assistant:")[-1].strip()
+        
         # 세션 스테이트에 대화 내용 업데이트
         st.session_state["past_user_inputs"].append(user_input)
         st.session_state["generated_responses"].append(bot_text)