Update app.py

app.py

@@ -25,11 +25,10 @@ def analyze(user_message, assistant_reply):
         {"role": "assistant", "content": assistant_reply},
     ]
 
-    input_ids = tokenizer.apply_chat_template(
-        messages, tokenize=True, return_tensors="pt"
-    ).to(DEVICE)
+    text = tokenizer.apply_chat_template(messages, tokenize=False)
+    input_ids = tokenizer(text, return_tensors="pt").input_ids.to(DEVICE)
 
-    tokens = tokenizer.convert_ids_to_tokens(input_ids[0])
+    tokens = tokenizer.convert_ids_to_tokens(input_ids[0].tolist())
 
     with torch.no_grad():
         _, _, values = model(input_ids)