sync ms

app.py

@@ -9,6 +9,20 @@ from bs4 import BeautifulSoup
 
 V_TO_SPLIT = {"IMAGENET1K_V1": "train", "IMAGENET1K_V2": "test"}
 
+EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
+
+EN2ZH = {
+    "ImageNet version": "ImageNet 版本",
+    "Clean cache": "清理缓存",
+    "Status": "状态栏",
+    "Download JSON lines": "下载 jsonl",
+    "Preview": "预览",
+}
+
+
+def _L(en_txt: str):
+    return en_txt if EN_US else f"{en_txt} ({EN2ZH[en_txt]})"
+
 
 def parse_url(url: str):
     response = requests.get(url)
@@ -120,7 +134,7 @@ def sync(subset: str):
             os.remove(cache_json)
 
         if os.path.exists(cache_json):
-            raise Exception(f"Failed to clean {cache_json}")
+            raise FileExistsError(f"Failed to clean {cache_json}")
 
     except Exception as e:
         status = f"{e}"
@@ -133,18 +147,18 @@ if __name__ == "__main__":
         with gr.Row():
             with gr.Column():
                 subset_opt = gr.Dropdown(
-                    label="ImageNet version",
+                    label=_L("ImageNet version"),
                     choices=["IMAGENET1K_V1", "IMAGENET1K_V2"],
                     value="IMAGENET1K_V1",
                 )
-                sync_btn = gr.Button("Clean cache")
+                sync_btn = gr.Button(_L("Clean cache"))
 
             with gr.Column():
-                status_bar = gr.Textbox(label="Status", show_copy_button=True)
-                dld_file = gr.File(label="Download JSON lines")
+                status_bar = gr.Textbox(label=_L("Status"), show_copy_button=True)
+                dld_file = gr.File(label=_L("Download JSON lines"))
 
         with gr.Row():
-            data_frame = gr.Dataframe(label="Preview")
+            data_frame = gr.Dataframe(label=_L("Preview"))
 
         subset_opt.change(
             infer,

requirements.txt

@@ -1,2 +1,2 @@
-bs4
-pandas
+pandas
+beautifulsoup4