updated task name
Browse files
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 "
|