Update app.py

app.py

@@ -6,9 +6,11 @@ model_name = "MajorJalud/Qwen2.5-0.5B-Instruct-Gensyn-Swarm-reptilian_strong_gul
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+
 # Function to handle the chat
 def chat(message, history):
     # Add a prompt to make the model act like a helpful assistant
+    inputs = tokenizer("You are a helpful assistant. User: " + message, return_tensors="pt")
     prompt = f"You are a helpful assistant. User: {message} Assistant: "
     inputs = tokenizer(prompt, return_tensors="pt")
     # Generate a response