Update app.py

app.py

@@ -43,7 +43,14 @@ def chat_function(message, history):
         
     messages.append({"role": "user", "content": message})
     
-    inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt")
+    # Fix: return_dict=False forces the tokenizer to deliver pure tensor arrays to the generation layer
+    inputs = tokenizer.apply_chat_template(
+        messages, 
+        add_generation_prompt=True, 
+        return_tensors="pt",
+        return_dict=False
+    )
+    
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
     
     generation_kwargs = dict(
@@ -61,6 +68,7 @@ def chat_function(message, history):
         partial_text += new_text
         yield partial_text
 
+
 demo = gr.ChatInterface(
     fn=chat_function,
     title="🤖 Cydercoder Qwen 3B AI Chatbot",