Update app.py

app.py

@@ -281,12 +281,13 @@ if st.button("Generate"):
                 
             elif task == "Translation" or "<translate>" in wrapped_input:
                 n_sentences = len(re.split(r"\.|\n", user_input))
-                print("split for translation: ",re.split(r"\.|\n", generated_text)[:n_sentences])
+                print("split for translation: ", n_sentences, re.split(r"\.|\n", generated_text)[:n_sentences], re.split(r"\.|\n", user_input))
                 generated_text = ".".join(re.split(r"\.|\n", generated_text)[:n_sentences])
 
             elif task == "Question Generation" or "Question Generation:" in sample_text:
                 if "?" in generated_text:
-                    generated_text = "?".join(re.split(r"\?", generated_text)[:-1])
+                    generated_text = "?".join(re.split(r"\?", generated_text)[:-1]) + "?"
+                    
                     
             full_output = st.empty()