fix error

app.py

@@ -32,12 +32,21 @@ def start():
 
     return "Model loaded and ready!"
 
-def send_message(input_text):
+def send_message(message, history):
     global tokenizer, model
+    # Add the user's message to the history
+    history.append(("User", message))
+    
+    # Generate the model's response
+    input_text = " ".join([msg for _, msg in history])
     input_ids = tokenizer(input_text, return_tensors='pt').input_ids.cuda()
     output = model.generate(inputs=input_ids, max_new_tokens=50)
     generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
-    return generated_text
+    
+    # Add the model's response to the history
+    history.append(("Bot", generated_text))
+    
+    return history, history
 
 with gr.Blocks() as demo:
     gr.Markdown("# Chat with the Model")
@@ -45,12 +54,12 @@ with gr.Blocks() as demo:
     start_button = gr.Button("Start Model")
     status_text = gr.Textbox(label="Status")
     
-    start()
+    start_button.click(start, inputs=None, outputs=status_text)
     
-    chatbox = gr.Chatbot()
+    chatbot = gr.Chatbot()
     message = gr.Textbox(label="Your Message")
     send_button = gr.Button("Send")
     
-    send_button.click(send_message, inputs=message, outputs=chatbox)
+    send_button.click(send_message, inputs=[message, chatbot], outputs=[chatbot, chatbot])
     
 demo.launch()

requirements.txt

@@ -1,3 +1,4 @@
 accelerate
 bitsandbytes
 transformers
+spaces