Upload 2 files

app.py

@@ -3039,7 +3039,7 @@ _initial_version = _initial_versions[0] if _initial_versions else None
 _initial_ckpt_labels = _ckpt_labels(_initial_version) if _initial_version else []
 _mode_keys = list(MODES.keys())
 
-with gr.Blocks(title="CompactAI Models", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="CompactAI Models") as demo:
     gr.Markdown(
         "# CompactAI — TinyMemoryLM\n"
         "Tiny recurrent-depth language models from [CompactAI-O](https://huggingface.co/CompactAI-O)."
@@ -3068,7 +3068,7 @@ with gr.Blocks(title="CompactAI Models", theme=gr.themes.Soft()) as demo:
                 c_use_custom, c_temp, c_topk, c_topp, c_minp, c_rep, c_ng, c_cap, c_lp, c_maxt, c_ctx, c_raw = _advanced_block()
 
             with gr.Column(scale=3):
-                chatbot = gr.Chatbot(label="Conversation", height=500, type="messages")
+                chatbot = gr.Chatbot(label="Conversation", height=500)
                 with gr.Row():
                     msg_box = gr.Textbox(placeholder="Type a message…", show_label=False, scale=5)
                     send_btn = gr.Button("Send", variant="primary", scale=1)
@@ -3148,4 +3148,4 @@ with gr.Blocks(title="CompactAI Models", theme=gr.themes.Soft()) as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Soft())

requirements.txt

@@ -1,4 +1,4 @@
-gradio>=4.44.0
+gradio>=6.0.0
 torch>=2.2.0
 huggingface_hub>=0.24.0
 datasets>=2.20.0