Update app.py

app.py

@@ -5,6 +5,7 @@ from tokenizers import Tokenizer
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file as load_safetensors
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # ----------------------------
 # 🔧 Model versions configuration
 # ----------------------------
@@ -41,7 +42,6 @@ config = {
     "tokenizer_path": "beeper.tokenizer.json"
 }
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Global model and tokenizer variables
 infer = None
@@ -185,7 +185,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
     
     # Chat interface
-    chatbot = gr.Chatbot(label="Chat with Beeper", type="messages", height=400)
+    chatbot = gr.Chatbot(label="Chat with Beeper", type="tuples", height=400)
     msg = gr.Textbox(label="Message", placeholder="Type your message here...")
     
     with gr.Row():
@@ -214,6 +214,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     # Handle chat
     def respond(message, chat_history, model_version, temperature, top_k, top_p):
+        if not chat_history:
+            chat_history = []
         response = beeper_reply(message, chat_history, model_version, temperature, top_k, top_p)
         chat_history.append([message, response])
         return "", chat_history