ck46 commited on
Commit
3309d53
1 Parent(s): 9d9f98b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,12 +15,12 @@ def preprocess_text(text):
15
  # Add a model selector to the sidebar
16
  q_model = st.sidebar.selectbox(
17
  'Select Question Generation Model',
18
- ('ck46/t5-base-squad-qa-qg', 'ck46/t5-small-squad-qa-qg')
19
  )
20
 
21
  a_model = st.sidebar.selectbox(
22
  'Select Answer Extraction Model',
23
- ('ck46/t5-base-squad-qa-qg', 'ck46/t5-small-squad-qa-qg')
24
  )
25
 
26
  st.header('Question-Answer Generation')
 
15
  # Add a model selector to the sidebar
16
  q_model = st.sidebar.selectbox(
17
  'Select Question Generation Model',
18
+ ('ck46/t5-base-hotpot-qa-qg', 'ck46/t5-small-hotpot-qa-qg')
19
  )
20
 
21
  a_model = st.sidebar.selectbox(
22
  'Select Answer Extraction Model',
23
+ ('ck46/t5-base-hotpot-qa-qg', 'ck46/t5-small-hotpot-qa-qg')
24
  )
25
 
26
  st.header('Question-Answer Generation')