BeardedMonster commited on
Commit
25ba82d
·
verified ·
1 Parent(s): 7b533d4
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -283,6 +283,9 @@ if st.button("Generate"):
283
  n_sentences = len(re.split(r"\.|\n", user_input))
284
  print("split for translation: ",re.split(r"\.|\n", generated_text)[:n_sentences])
285
  generated_text = ".".join(re.split(r"\.|\n", generated_text)[:n_sentences])
 
 
 
286
 
287
  full_output = st.empty()
288
 
 
283
  n_sentences = len(re.split(r"\.|\n", user_input))
284
  print("split for translation: ",re.split(r"\.|\n", generated_text)[:n_sentences])
285
  generated_text = ".".join(re.split(r"\.|\n", generated_text)[:n_sentences])
286
+
287
+ elif task == "Question Generation":
288
+ generated_text = "?".join(re.split("?", generated_text)[:-1])
289
 
290
  full_output = st.empty()
291