Update app.py

app.py

@@ -28,7 +28,7 @@ def chat_with_model(message, history, model_choice):
     tokenizer, model = load_model(model_choice)
     device = model.device
     
-    prompt = f"User: {message}\nAssistant:"
+    prompt = message
     
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
     inputs = tokenizer(prompt, return_tensors="pt").to(device)
@@ -36,12 +36,10 @@ def chat_with_model(message, history, model_choice):
     generation_kwargs = dict(
         **inputs,
         streamer=streamer,
-        max_new_tokens=512,
+        max_new_tokens=1024,
         temperature=0.7,
         top_p=0.9,
         do_sample=True,
-        stop_strings=["User:", "Assistant:"],
-        tokenizer=tokenizer
     )
     
     thread = Thread(target=model.generate, kwargs=generation_kwargs)
@@ -49,8 +47,6 @@ def chat_with_model(message, history, model_choice):
     
     partial_text = ""
     for new_text in streamer:
-        if "User:" in new_text:
-            break
         partial_text += new_text
         yield [
             {"role": "user", "content": message},
@@ -73,32 +69,28 @@ def create_demo():
 
     with gr.Blocks(theme=custom_theme, title="Creeper AI Chatbot") as demo:
         gr.Markdown("# 🌿 Creeper AI Chatbot")
-        gr.Markdown("*Low-latency Liquid Intelligence*")
         
-        with gr.Row():
-            model_choice = gr.Dropdown(
-                label="Model Core",
-                choices=list(MODEL_NAMES.keys()),
-                value="LFM 1.2B",
-                interactive=True
-            )
+        model_choice = gr.Dropdown(
+            label="Select Model",
+            choices=list(MODEL_NAMES.keys()),
+            value="LFM 1.2B"
+        )
         
         chatbot = gr.Chatbot(
-            label="Neural Interface",
+            label="Raw AI Output",
             type="messages",
-            height=500,
-            show_copy_button=True
+            height=500
         )
         
         with gr.Row():
             msg = gr.Textbox(
-                label="Input Command",
-                placeholder="Message Creeper AI...",
+                label="Direct Input",
+                placeholder="Type exactly what you want to send...",
                 scale=4
             )
             submit_btn = gr.Button("Send", variant="primary", scale=1)
         
-        clear = gr.Button("Reset Memory (Wipe)")
+        clear = gr.Button("Clear")
         
         def add_user_message(user_message):
             return "", [{"role": "user", "content": user_message}]