BeardedMonster commited on
Commit
1f545b9
·
verified ·
1 Parent(s): 25ba82d

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -284,7 +284,7 @@ if st.button("Generate"):
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()
 
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" or "Question Generation:" in sample_text:
288
  generated_text = "?".join(re.split("?", generated_text)[:-1])
289
 
290
  full_output = st.empty()