Update app.py

app.py

@@ -3,14 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStream
 import torch
 from threading import Thread
 MODEL_NAMES = {
-    "LFM2-350M": "LiquidAI/LFM2-350M",                              # LFM2   - 350M 
-    "LFM2-700M": "LiquidAI/LFM2-700M",                              # LFM2   - 700M
-    "LFM2-1.2B": "LiquidAI/LFM2-1.2B",                              # LFM2   - 1.2B
-    "LFM2.5-1.2B": "LiquidAI/LFM2.5-1.2B-Instruct",                 # LFM2.5 - 1.2B
-    "LFM2.5-1.2B-Thinking": "LiquidAI/LFM2.5-1.2B-Thinking",        # LFM2.5 - 1.2B (Thinking)
-    "LFM2-2.6B": "LiquidAI/LFM2-2.6B",                              # LFM2   - 2.6B
-    "LFM2-8B-A1B": "LiquidAI/LFM2-8B-A1B",                          # LFM2   - 8B-A1B
-    "LFM2-24B-A3B": "LiquidAI/LFM2-24B-A2B",                        # LFM2   - 24B-A3B
+    "dqnCode v0.2 1.5B": "DQN-Labs/dqnCode-v0.2-1.5B-HF", 
+    "dqnCode v0.3 1.2B": "DQN-Labs/dqnCode-v0.3-1.2B-MLX-4bit",
 }
 model_cache = {}
 def load_model(model_key):
@@ -55,19 +49,19 @@ def chat_with_model(message, history, model_choice):
             {"role": "assistant", "content": partial_text},
         ]
 def create_demo():
-    with gr.Blocks(title="LiquidAI Chat Playground") as demo:
-        gr.Markdown("## 💧 LiquidAI Chat Playground")
+    with gr.Blocks(title="DQN Labs Chat") as demo:
+        gr.Markdown("## DQN Labs Chat")
         model_choice = gr.Dropdown(
             label="Select Model",
             choices=list(MODEL_NAMES.keys()),
-            value="LFM2-1.2B"
+            value="dqnCode v0.2 1.5B"
         )
         chatbot = gr.Chatbot(
-            label="Chat with LiquidAI",
+            label="Chat with DQN!",
             type="messages",
             height=450
         )
-        msg = gr.Textbox(label="Your message", placeholder="Type something...")
+        msg = gr.Textbox(label="Your message", placeholder="Type away...")
         clear = gr.Button("Clear")
         def add_user_message(user_message, chat_history):
             chat_history = chat_history + [{"role": "user", "content": user_message}]