Update app.py

app.py

@@ -5,6 +5,12 @@ import random
 import pandas as pd
 from huggingface_hub import login
 
+# 下載判決書資料集
+login(token = os.environ['HUB_TOKEN'])
+use_auth_token=os.environ['HUB_TOKEN'] # 下載判決書資料集所需要的 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) 當中作為斷詞字元的空白字元，以及每句開頭的「ㄧ、」。
@@ -114,9 +120,4 @@ with gr.Blocks() as demo:
 #        btn3.click(download_file, inputs=[result, title], outputs=[])
 
 if __name__ == "__main__":
-    # 下載判決書資料集
-    login(token = os.environ['HUB_TOKEN'])
-    use_auth_token=os.environ['HUB_TOKEN'] # 下載判決書資料集所需要的 token。
-    dataset = load_dataset("jslin09/Fraud_Case_Verdicts", use_auth_token=use_auth_token, revision="main")
-    dataset = dataset.map(remove_space)
     demo.launch(share=True) # 在遠端啟動時，需要 share=True 。