Update app.py

app.py

@@ -302,14 +302,15 @@ if st.button("Generate"):
             elif task == "Topic Classification" or "<topic>" in wrapped_input:
                 generated_text  = generated_text[:15]
                 # print("split", generated_text.split(" ")[0],  re.split(r"\.|\n|\*\*|\*", generated_text)[0], generated_text.split(" "))
-                generated_text = re.split(r"\.|\n|\*\*|\*", generated_text)[0] 
+                generated_text = re.split(r"\.|\n|\*\*|\*", generated_text)[0] + "."
                 generated_text = asyncio.run(assign_topic(generated_text))
                 
             elif task == "Translation" or "<translate>" in wrapped_input:
                 # print("split for translation: ", n_sentences, re.split(r"\.|\n", generated_text)[:n_sentences])
                 n_sentences, split_= count_sentences(initial_input)
-                print(n_sentences,split_)
-                generated_text = ". ".join(re.split(r"\.|\n", generated_text)[:n_sentences]) + "."
+                print(n_sentences, split_)
+                _, generated_text = count_sentences(generated_text)
+                generated_text = ". ".join(generated_text)[:n_sentences]) + "."
 
             elif task == "Question Generation" or "Question Generation:" in sample_text:
                 if "?" in generated_text: