Update app.py

app.py

@@ -20,6 +20,7 @@ fpretrain_lst = ["None", "-scsmall", "-scmedium", "-sclarge"]
 finetune_lst = ["-squad", "-scqa1", "-scqa2"]
 
 # 为每个选项创建下拉菜单
+st.markdown(f'bert', unsafe_allow_html=True)
 size = st.selectbox("Choose a model size:", size_lst)
 cased = st.selectbox("Whether distinguish upper and lowercase letters:", cased_lst)
 fpretrain = st.selectbox("Further pretrained on a solar cell corpus:", fpretrain_lst)
@@ -33,7 +34,7 @@ else:
 
 # 显示用户选择的模型
 st.write(f"Your selected model:")
-st.markdown(f'##### {model}', unsafe_allow_html=True)
+st.markdown(f'###### {model}', unsafe_allow_html=True)
 
 # 加载问答模型
 pipe = pipeline("question-answering", model=model)
@@ -116,6 +117,7 @@ if st.button('Extract the property'):
 # Transformation Algorithm
 ##########
 st.markdown('## 🖥️ QA Dataset Auto Generation', unsafe_allow_html=True)
+st.markdown('### Input', unsafe_allow_html=True)
 
 cde_lst = ["./CDE_properties.jsonl"]
 paper_lst = ["./reference_paper.json"]
@@ -139,6 +141,8 @@ with open(st.session_state['paper'],'r+', encoding = "utf-8") as f:
     json_string = json.dumps(json_data, indent=4)
     st.text_area("", value=json_string, height=200)
 
+st.markdown('### Output', unsafe_allow_html=True)
+
 st.write("To be updated")