Update app.py

app.py

@@ -1,17 +1,36 @@
 import streamlit as st
 from transformers import pipeline
 
-# 定义一个包含模型名称的列表
-models = [
-    "bert-base-uncased",
-    "distilbert-base-uncased",
-    "roberta-base",
-    "gpt2",
-    "ZongqianLi/bert-base-cased-scqa2"  # 你可以在这里添加更多模型
+size_lst = [
+    "-base",
+    "-large"
 ]
 
-# 使用st.selectbox创建一个下拉选择框，让用户从列表中选择一个模型
-model = st.selectbox("Choose a model:", models)
+cased_lst = [
+    "-cased",
+    "-uncased"
+]
+
+fpretrain_lst = [
+    "None",
+    "-scsmall",
+    "-scmedium",
+    "-sclarge"
+]
+
+finetune_lst = [
+    "-squad",
+    "-scqa1",
+    "-scqa2"
+]
+
+size = st.selectbox("Choose a model size: ", size_lst)
+cased = st.selectbox("Whether distinguish upper and lowercase letters: ", size_lst)
+fpretrain = st.selectbox("Further pretrained on a solar cell corpus: ", fpretrain_lst)
+finetune = st.selectbox("Finetuned on a QA dataset: ", finetune_lst)
+
+if fpretrain == "None":
+    model = "".join("ZongqianLi/bert-", size, cased, fpretrain, finetune)
 
 # 显示用户当前选择的模型
 st.write(f"You selected: {model}")