[minor] setup

app.py

@@ -30,9 +30,9 @@ class ModelManager:
     
     def __init__(
         self,
-        single_model_name: str = "Qwen/Qwen2.5-0.5B-Instruct",
-        t2t_context_model: str = "Qwen/Qwen3-0.6B",
-        t2t_answer_model: str = "Qwen/Qwen2.5-0.5B-Instruct",
+        single_model_name: str = "Qwen/Qwen3-0.6B",
+        t2t_context_model: str = "Qwen/Qwen2.5-0.5B-Instruct",
+        t2t_answer_model: str = "Qwen/Qwen3-0.6B",
         c2c_checkpoint_path: str = "local/checkpoints/qwen3_0.6b+qwen2.5_0.5b_Fuser",
         device: str = "auto"
     ):
@@ -502,7 +502,7 @@ D. Whether plants have interests.""",
             # C2C column
             with gr.Column():
                 gr.Markdown("### Cache-to-Cache Communication")
-                gr.Markdown(f"*{model_manager.c2c_base_model} → {model_manager.c2c_teacher_model}*")
+                gr.Markdown(f"*{model_manager.c2c_teacher_model} → {model_manager.c2c_base_model}*")
                 c2c_output = gr.Textbox(
                     label="",
                     lines=18,
@@ -553,7 +553,21 @@ def main():
     print("=" * 60)
     
     # Initialize models
-    model_manager = ModelManager()
+    # C2C-S: qwen3_0.6b+qwen2.5_0.5b_Fuser
+    context_model_name = "Qwen/Qwen2.5-0.5B-Instruct"
+    c2c_checkpoint_path = "local/checkpoints/qwen3_0.6b+qwen2.5_0.5b_Fuser"
+
+    # C2C-L: qwen3_0.6b+qwen2.5_0.5b_Fuser_large
+    # context_model_name = "Qwen/Qwen3-4B-Base"
+    # c2c_checkpoint_path = "local/checkpoints/qwen3_0.6b+qwen3_4b_base_Fuser"
+
+    answer_model_name = "Qwen/Qwen3-0.6B"
+    model_manager = ModelManager(
+        single_model_name=answer_model_name,
+        t2t_context_model=context_model_name,
+        t2t_answer_model=answer_model_name,
+        c2c_checkpoint_path=c2c_checkpoint_path
+    )
     
     # Create and launch demo
     demo = create_demo(model_manager)
@@ -565,9 +579,9 @@ def main():
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True,
+        share=False,
         show_error=True,
-        # ssr_mode=False
+        ssr_mode=False
     )