sunwaee commited on
Commit
92aedc8
1 Parent(s): 73d960a

updated task name

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,7 +60,7 @@ download_models(ids)
60
  # Task selection
61
 
62
  left, right = st.columns([4, 2])
63
- task = left.selectbox('', options=['Questions/Answers Generation', 'Question Answering', 'Question Generation'],
64
  help='Choose the task you want to try out')
65
 
66
  # Model selection
@@ -68,7 +68,7 @@ model_path = right.selectbox('', options=[k for k in ids], index=0, help='Model
68
  model = load_model(model_path=f"model/{model_path}.ckpt")
69
  right.write(model.device)
70
 
71
- if task == 'Questions/Answers Generation':
72
  # Input area
73
  inputs = st.text_area('Context:', value="A few years after the First Crusade, in 1107, the Normans under "
74
  "the command of Bohemond, Robert\'s son, landed in Valona and "
 
60
  # Task selection
61
 
62
  left, right = st.columns([4, 2])
63
+ task = left.selectbox('', options=['Questions/Answers Pairs Generation', 'Question Answering', 'Question Generation'],
64
  help='Choose the task you want to try out')
65
 
66
  # Model selection
 
68
  model = load_model(model_path=f"model/{model_path}.ckpt")
69
  right.write(model.device)
70
 
71
+ if task == 'Questions/Answers Pairs Generation':
72
  # Input area
73
  inputs = st.text_area('Context:', value="A few years after the First Crusade, in 1107, the Normans under "
74
  "the command of Bohemond, Robert\'s son, landed in Valona and "