2 en

app.py

@@ -7,13 +7,14 @@ from modelscope import snapshot_download
 from insectid import InsectDetector, InsectIdentifier
 
 MODEL_DIR = snapshot_download(
-    "Genius-Society/insecta", cache_dir="./insectid/__pycache__"
+    "Genius-Society/insecta",
+    cache_dir="./insectid/__pycache__",
 )
 
 
 def infer(filename: str):
     if not filename:
-        None, "请上传图片 Please upload a picture"
+        None, "Please upload a picture"
 
     detector = InsectDetector()
     identifier = InsectIdentifier()
@@ -27,7 +28,7 @@ def infer(filename: str):
     image_for_draw = image.copy()
     image_height, image_width = image.shape[:2]
     boxes, confs, classes = detector.detect(image)
-    text = "未知"
+    text = "Unknown"
     for box, _, _ in zip(boxes, confs, classes):
         box = box.astype(np.int32)
         box_width = box[2] - box[0] + 1
@@ -40,8 +41,7 @@ def infer(filename: str):
         print(results[0])
         prob = results[0]["probability"]
         if prob >= 0.10:
-            text = "{} {}: {:.2f}%".format(
-                results[0]["chinese_name"],
+            text = "{}: {:.2f}%".format(
                 results[0]["latin_name"],
                 100.0 * results[0]["probability"],
             )
@@ -72,14 +72,12 @@ if __name__ == "__main__":
     with gr.Blocks() as demo:
         gr.Interface(
             fn=infer,
-            inputs=gr.Image(
-                label="上传昆虫照片 Upload insect picture", type="filepath"
-            ),
+            inputs=gr.Image(label="Upload insect picture", type="filepath"),
             outputs=[
-                gr.Image(label="识别结果 Recognition result"),
-                gr.Textbox(label="最可能的物种 Best match", show_copy_button=True),
+                gr.Image(label="Recognition result"),
+                gr.Textbox(label="Best match", show_copy_button=True),
             ],
-            title="图像文件格式支持 PNG, JPG, JPEG 和 BMP, 且文件大小不超过 10M<br>Image file format support PNG, JPG, JPEG and BMP, and the file size does not exceed 10M.",
+            title="Image file format support PNG, JPG, JPEG and BMP, and the file size does not exceed 10M.",
             examples=[
                 f"{MODEL_DIR}/examples/butterfly.jpg",
                 f"{MODEL_DIR}/examples/beetle.jpg",