Update app.py

app.py

@@ -181,37 +181,18 @@ def generate_text_better_sampling(model, prompt, max_len=100, max_gen=98, top_k=
 
         yield decoded_text
 
+nickname = "사용자"
 
-import gradio as gr
-
-nickname = "사용자1"
-
-def respond(message, history):
+def respond(message, chat_history):
     message = message.replace("@사용자1@", nickname)
-    history = history or []
-    bot_response = ""
-
+    response = ""
     for partial in generate_text_better_sampling(model, message):
-        bot_response = partial
-        updated_history = history + [(nickname, message), ("봇", bot_response)]
-        yield updated_history
-
-with gr.Blocks() as demo:
-    state = gr.State([])
-    chat_display = gr.Markdown()
-    user_input = gr.Textbox(placeholder="메시지를 입력하세요...", show_label=False)
-
-    def format_history(history):
-        return "\n\n".join([f"**{who}**: {what}" for who, what in history])
-
-    def update(history):
-        return format_history(history), history
-
-    def stream_response(message, history):
-        for updated_history in respond(message, history):
-            yield format_history(updated_history), updated_history
+        response = partial
+        yield response
 
-    user_input.submit(stream_response, inputs=[user_input, state], outputs=[chat_display, state])
-    user_input.submit(lambda: "", None, user_input)  # 입력창 초기화
+chat = gr.ChatInterface(
+    fn=respond,
+    title="InteractGPT",
+)
 
-demo.launch()
+chat.launch()