Update app.py

app.py

@@ -1,12 +1,29 @@
 import streamlit as st
 from transformers import pipeline
 
+import streamlit as st
+from st_chat_message import message
+
+message("Hello world!", is_user=True)
+message("Hi")
+
+st.markdown("# Auto-generating Question-Answering Datasets with Domain-Specific Knowledge for Language Models in Scientific Tasks", unsafe_allow_html=True)
+
+##########
+# Transformation Algorithm
+##########
+st.markdown('## QA Dataset Auto Generation', unsafe_allow_html=True)
+
+
+
+
+
 ##########
 # Property Extraction
 ##########
-st.markdown('# Property Extraction', unsafe_allow_html=True)
+st.markdown('## Property Extraction', unsafe_allow_html=True)
 
-st.markdown('## Select a model: ', unsafe_allow_html=True)
+st.markdown('### Select a model: ', unsafe_allow_html=True)
 
 # 定义模型配置选项
 size_lst = ["-base", "-large"]
@@ -32,7 +49,7 @@ st.write(f"Your selected model: {model}")
 # 加载问答模型
 pipe = pipeline("question-answering", model=model)
 
-st.markdown('## Input a paper: ', unsafe_allow_html=True)
+st.markdown('### Input a paper: ', unsafe_allow_html=True)
 
 # 设置默认的问题和上下文
 default_property = "FF"