Update app.py

app.py

@@ -51,8 +51,8 @@ st.header("Contract Understanding Atticus Dataset (CUAD) Demo")
 st.write("Based on https://github.com/marshmellow77/cuad-demo")
 
 
-question = st.selectbox('Choose one of the 41 queries from the CUAD dataset:', questions)
-# paragraph = st.text_area(label="Contract")
+selected_question = st.selectbox('Choose one of the 41 queries from the CUAD dataset:', questions)
+question_set = [questions[0], selected_question]
 
 contract_type = st.radio("Select Contract", ("Sample Contract", "New Contract"))
 if contract_type == "Sample Contract":
@@ -66,5 +66,5 @@ else:
 Run_Button = st.button("Run", key=None)
 if Run_Button == True and not len(contract)==0 and not len(question)==0:
 
-	prediction = run_prediction(question, contract, 'akdeniz27/roberta-base-cuad')
+	prediction = run_prediction(question_set, contract, 'akdeniz27/roberta-base-cuad')
 	st.write("Answer: " + prediction.strip())