Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
-model_id = "openai-community/gpt2"
+model_id = "cody82/unitrip"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
@@ -35,12 +35,17 @@ def respond(user_message, history):
     history.append((user_message, generated_text))
     return history, history
 
+def clear_textbox():
+    return ""
+
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     message = gr.Textbox(placeholder="Введите вопрос...")
     state = gr.State([])
 
+    # При отправке сообщения вызываем respond, обновляем чат и состояние
     message.submit(respond, inputs=[message, state], outputs=[chatbot, state])
-    message.submit(lambda: "", None, message)  # очистка поля ввода
+    # Очищаем поле ввода после отправки
+    message.submit(clear_textbox, inputs=[], outputs=[message])
 
 demo.launch(share=True)