Update main.py

main.py

@@ -65,18 +65,13 @@ async def generate_response(data: InputData) -> Dict[str, Any]:
                 cleaned_key = key.replace("###", "")
                 cleaned_response[cleaned_key] = value
 
-            """for i, text in enumerate(cleaned_response["New response"]):
-                if i <= 2:
-                    sentences = tokenizer.tokenize(text)
-                    if sentences:
-                        cleaned_response["New response"][i] = sentences[0]
-                else:
-                    del cleaned_response["New response"][i]"""
-
             strings = ""
                 
-            for _, text in enumerate(cleaned_response["New response"]):
-                strings += " " + text
+            for i, text in enumerate(cleaned_response["New response"]):
+                if i != len(cleaned_response["New response"]) - 1:
+                    strings += text + " "
+                else:
+                    strings += text
 
             sentences = tokenizer.tokenize(strings)