coding

app.py

@@ -7,8 +7,9 @@ import random
 
 # 使用 進擊的巨人 数据集
 try:
+    converter = pipeline("translation_zh_tw_zh_cn")
     dataset = load_dataset("rorubyy/attack_on_titan_wiki_chinese")
-    answer_list = [example["Answer"] for example in dataset["train"]]
+    answer_list = [converter(example["Answer"])[0]["translation_text"] for example in dataset["train"]]
 
 except Exception as e:
     st.error(f"读取数据集失败：{e}")
@@ -67,14 +68,18 @@ with col2:
     temperature = st.number_input("temperature", value=1.0)
     max_length = st.number_input("max_length", value=1024)
 
+st.divider()
+
 col3, col4 = st.columns(2)
 with col3:
-    if st.button("随机"):
+    if st.button("使用原数据集中的随机问题"):
         dataset_size = len(dataset["train"])
         random_index = random.randint(0, dataset_size - 1)
         # 读取随机问题
         random_question = dataset["train"][random_index]["Question"]
+        random_question = converter(random_question)[0]["translation_text"]
         origin_answer = dataset["train"][random_index]["Answer"]
+        origin_answer = converter(origin_answer)[0]["translation_text"]
         print('[]' + str(random_index) + '/' + str(dataset_size) + ']random_question: ' + random_question)
         print('origin_answer: ' + origin_answer)
 
@@ -90,7 +95,7 @@ with col3:
 
 with col4:
     question = st.text_area("请输入问题", "Gemma 有哪些特点？")
-    if st.button("提交"):
+    if st.button("提交输入的问题"):
         if not question:
             st.warning("请输入问题！")
         else: