Update app.py

app.py

@@ -11,7 +11,6 @@ from utils import is_url, download_file, get_jpg_files, MODEL_DIR
 
 TMP_DIR = "./__pycache__"
 
-
 @dataclass
 class Cfg:
     detector_weights: str
@@ -21,7 +20,6 @@ class Cfg:
     disable_faces: bool = False
     draw: bool = True
 
-
 class ValidImgDetector:
     predictor = None
 
@@ -39,7 +37,7 @@ class ValidImgDetector:
         mode: str,
         predictor: Predictor,
     ) -> np.ndarray:
-        # input is rgb image, output must be rgb too
+        # input is RGB image, output must be RGB too
         predictor.detector.detector_kwargs["conf"] = score_threshold
         predictor.detector.detector_kwargs["iou"] = iou_threshold
         if mode == "Use persons and faces":
@@ -79,8 +77,8 @@ def infer(photo: str):
         photo = download_file(photo, f"{TMP_DIR}/download.jpg")
 
     detector = ValidImgDetector()
-    if not photo or not os.path.exists(photo) or imghdr.what(photo) == None:
-        return None, None, None, "请正确输入图片 Please input image correctly"
+    if not photo or not os.path.exists(photo) or imghdr.what(photo) is None:
+        return None, None, None, "Please input the image correctly"
 
     return detector.valid_img(photo)
 
@@ -88,30 +86,30 @@ def infer(photo: str):
 if __name__ == "__main__":
     with gr.Blocks() as iface:
         warnings.filterwarnings("ignore")
-        with gr.Tab("上传模式 Upload Mode"):
+        with gr.Tab("Upload Mode"):
             gr.Interface(
                 fn=infer,
-                inputs=gr.Image(label="上传照片 Upload Photo", type="filepath"),
+                inputs=gr.Image(label="Upload Photo", type="filepath"),
                 outputs=[
-                    gr.Image(label="检测结果 Detection Result", type="numpy"),
-                    gr.Textbox(label="存在儿童 Has Child"),
-                    gr.Textbox(label="存在女性 Has Female"),
-                    gr.Textbox(label="存在男性 Has Male"),
+                    gr.Image(label="Detection Result", type="numpy"),
+                    gr.Textbox(label="Has Child"),
+                    gr.Textbox(label="Has Female"),
+                    gr.Textbox(label="Has Male"),
                 ],
                 examples=get_jpg_files(f"{MODEL_DIR}/examples"),
                 allow_flagging="never",
                 cache_examples=False,
             )
 
-        with gr.Tab("在线模式 Online Mode"):
+        with gr.Tab("Online Mode"):
             gr.Interface(
                 fn=infer,
-                inputs=gr.Textbox(label="网络图片链接 Online Picture URL"),
+                inputs=gr.Textbox(label="Online Picture URL"),
                 outputs=[
-                    gr.Image(label="检测结果 Detection Result", type="numpy"),
-                    gr.Textbox(label="存在儿童 Has Child"),
-                    gr.Textbox(label="存在女性 Has Female"),
-                    gr.Textbox(label="存在男性 Has Male"),
+                    gr.Image(label="Detection Result", type="numpy"),
+                    gr.Textbox(label="Has Child"),
+                    gr.Textbox(label="Has Female"),
+                    gr.Textbox(label="Has Male"),
                 ],
                 allow_flagging="never",
             )