Update app.py

app.py

@@ -6,12 +6,6 @@ import random
 import pandas as pd
 from huggingface_hub import login
 
-# 下載判決書資料集
-use_auth_token = os.environ['HUB_TOKEN'] # 下載判決書資料集所需要的 token。
-login(token = os.environ['HUB_TOKEN'])
-dataset = load_dataset("jslin09/Fraud_Case_Verdicts", use_auth_token=use_auth_token, revision="main")
-dataset = dataset.map(remove_space)
-
 def remove_space(example):
     '''
     移除資料集當中「犯罪事實」欄 (Corpus-Delicti) 當中作為斷詞字元的空白字元，以及每句開頭的「ㄧ、」。
@@ -76,6 +70,11 @@ def gen_template(crime_descrip, element, tag):
     template = blurb + '\n' + descript + '\n' + element + '\n' + tag + '\n' + end
     return template
 
+# 下載判決書資料集
+use_auth_token = os.environ['HUB_TOKEN'] # 下載判決書資料集所需要的 token。
+login(token = os.environ['HUB_TOKEN'])
+dataset = load_dataset("jslin09/Fraud_Case_Verdicts", use_auth_token=use_auth_token, revision="main")
+dataset = dataset.map(remove_space)
 #random_selected = random_elements(dataset["train"])
 random_selected = random_next()
 court_name = random_selected[0]