Update main.py

main.py

@@ -38,7 +38,7 @@ def read_root(data: req):
   input_text = data.prompt
   
   # Tokenize the input text
-  input_ids = tokenizer.encode(history_string, input_text, return_tensors="pt")
+  input_ids = tokenizer.encode_plus(history_string, input_text, return_tensors="pt")
   
   # Generate output using the model
   output_ids = model.generate(input_ids, num_beams=5, no_repeat_ngram_size=2)