sync ms

app.py

@@ -1,12 +1,27 @@
 import os
 import torch
-import modelscope
-import huggingface_hub
 import gradio as gr
 from PIL import Image
 from torchvision.transforms import transforms
 
 EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
+
+if EN_US:
+    import huggingface_hub
+
+    MODEL_DIR = huggingface_hub.snapshot_download(
+        "Genius-Society/HEp2",
+        cache_dir="./__pycache__",
+    )
+
+else:
+    import modelscope
+
+    MODEL_DIR = modelscope.snapshot_download(
+        "Genius-Society/HEp2",
+        cache_dir="./__pycache__",
+    )
+
 ZH2EN = {
     "上传细胞图像": "Upload a cell picture",
     "状态栏": "Status",
@@ -20,18 +35,6 @@ def _L(zh_txt: str):
     return ZH2EN[zh_txt] if EN_US else zh_txt
 
 
-MODEL_DIR = (
-    huggingface_hub.snapshot_download(
-        "Genius-Society/HEp2",
-        cache_dir="./__pycache__",
-    )
-    if EN_US
-    else modelscope.snapshot_download(
-        "Genius-Society/HEp2",
-        cache_dir="./__pycache__",
-    )
-)
-
 TRANSLATE = {
     "Centromere": "着丝粒",
     "Golgi": "高尔基体",

requirements.txt

@@ -1,5 +1,2 @@
-torch==2.6.0+cu118
--f https://download.pytorch.org/whl/torch
-torchvision==0.21.0+cu118
--f https://download.pytorch.org/whl/torchvision
-modelscope[framework]==1.24.0
+torch
+torchvision