2 en

app.py

@@ -6,15 +6,14 @@ from torchvision.transforms import transforms
 from modelscope import snapshot_download
 
 MODEL_DIR = snapshot_download("Genius-Society/HEp2", cache_dir="./__pycache__")
-TRANSLATE = {
-    "Centromere": "着丝粒 Centromere",
-    "Golgi": "高尔基体 Golgi",
-    "Homogeneous": "同质 Homogeneous",
-    "NuMem": "记忆体 NuMem",
-    "Nucleolar": "核仁 Nucleolar",
-    "Speckled": "斑核 Speckled",
-}
-CLASSES = list(TRANSLATE.keys())
+CLASSES = [
+    "Centromere",
+    "Golgi",
+    "Homogeneous",
+    "NuMem",
+    "Nucleolar",
+    "Speckled",
+]
 
 
 def embeding(img_path: str):
@@ -34,13 +33,13 @@ def embeding(img_path: str):
 def infer(target: str):
     model = torch.load(f"{MODEL_DIR}/save.pt", map_location=torch.device("cpu"))
     if not target:
-        return None, "请上传细胞图片 Please upload a cell picture!"
+        return None, "Please upload a cell picture!"
 
     torch.cuda.empty_cache()
     input: torch.Tensor = embeding(target)
     output: torch.Tensor = model(input.unsqueeze(0))
     predict = torch.max(output.data, 1)[1]
-    return os.path.basename(target), TRANSLATE[CLASSES[predict]]
+    return os.path.basename(target), CLASSES[predict]
 
 
 if __name__ == "__main__":
@@ -51,14 +50,12 @@ if __name__ == "__main__":
     with gr.Blocks() as demo:
         gr.Interface(
             fn=infer,
-            inputs=gr.Image(
-                type="filepath", label="上传细胞图像 Upload a cell picture"
-            ),
+            inputs=gr.Image(type="filepath", label="Upload a cell picture"),
             outputs=[
-                gr.Textbox(label="图片名 Picture name", show_copy_button=True),
-                gr.Textbox(label="识别结果 Recognition result", show_copy_button=True),
+                gr.Textbox(label="Picture name", show_copy_button=True),
+                gr.Textbox(label="Recognition result", show_copy_button=True),
             ],
-            title="请上传 PNG 格式的 HEp2 细胞图片<br>It is recommended to upload HEp2 cell images in PNG format.",
+            title="It is recommended to upload HEp2 cell images in PNG format.",
             examples=example_imgs,
             flagging_mode="never",
             cache_examples=False,