Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 from datasets import ClassLabel
 from datasets import load_dataset
@@ -6,8 +7,8 @@ import pandas as pd
 from huggingface_hub import login
 
 # 下載判決書資料集
+use_auth_token = os.environ['HUB_TOKEN'] # 下載判決書資料集所需要的 token。
 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)