Update app.py

app.py

@@ -35,8 +35,8 @@ if input_text:
 
     # Tokenize the input text and history
     inputs = tokenizer.encode_plus(history_string, return_tensors="pt")
-    inputs["input_ids"] = torch.cat([inputs["input_ids"], torch.tensor([tokenizer.sep_token_id])], dim=-1)
-    inputs["attention_mask"] = torch.cat([inputs["attention_mask"], torch.tensor([1])], dim=-1)
+    inputs["input_ids"] = torch.cat([inputs["input_ids"], torch.tensor([[tokenizer.sep_token_id]])], dim=-1)
+    inputs["attention_mask"] = torch.cat([inputs["attention_mask"], torch.tensor([[1]])], dim=-1)
     inputs = tokenizer.encode_plus(input_text, return_tensors="pt", add_special_tokens=False)
     inputs["input_ids"] = torch.cat([inputs["input_ids"], inputs["input_ids"]], dim=-1)
     inputs["attention_mask"] = torch.cat([inputs["attention_mask"], inputs["attention_mask"]], dim=-1)