Update app_gradio.py

app_gradio.py

@@ -2,21 +2,24 @@
 # @Author: SWHL
 # @Contact: [email protected]
 import gradio as gr
+from PIL import Image
 
+img_path = "images/1.jpg"
 
-def welcome(name):
-    return f"Welcome to Gradio, {name}!"
 
+def test():
+    return Image.open(img_path)
 
-title_content = """
-    # Rapid⚡OCR
-    <p align="center">
-        <a href=""><img src="https://img.shields.io/badge/Python->=3.6-aff.svg"></a>
-        <a href=""><img src="https://img.shields.io/badge/OS-Linux%2C%20Win%2C%20Mac-pink.svg"></a>
-        <a href="https://pepy.tech/project/rapidocr"><img src="https://static.pepy.tech/personalized-badge/rapidocr?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=Downloads%20Ort"></a>
-        <a href="https://pypi.org/project/rapidocr/"><img alt="PyPI" src="https://img.shields.io/pypi/v/rapidocr"></a>
-    </p>
-"""
+
+example_images = [
+    "images/1.jpg",
+    "images/ch_en_num.jpg",
+    "images/air_ticket.jpg",
+    "images/car_plate.jpeg",
+    "images/train_ticket.jpeg",
+    "images/japan_2.jpg",
+    "images/korean_1.jpg",
+]
 
 custom_css = """
     body {font-family: body {font-family: 'Helvetica Neue', Helvetica;}
@@ -27,6 +30,7 @@ custom_css = """
     .example-button:hover {background-color: #FF4500;}
     .tall-radio .gr-radio-item {padding: 15px 0; min-height: 50px; display: flex; align-items: center;}
     .tall-radio label {font-size: 16px;}
+    .output-image, .input-image, .image-preview {height: 300px !important}
 """
 
 with gr.Blocks(
@@ -46,9 +50,46 @@ with gr.Blocks(
         </div>
     """
     )
-    inp = gr.Textbox(placeholder="What is your name?")
-    out = gr.Textbox()
-    inp.change(welcome, inp, out)
+    with gr.Row():
+        text_score = gr.Slider(
+            label="text_score",
+            minimum=0,
+            maximum=1.0,
+            value=0.5,
+            step=0.1,
+            info="文本识别结果是正确的置信度，值越大，显示出的识别结果更准确。存在漏检时，调低该值。取值范围：[0, 1.0]，默认值为0.5",
+        )
+        box_thresh = gr.Slider(
+            label="box_thresh",
+            minimum=0,
+            maximum=1.0,
+            value=0.5,
+            step=0.1,
+            info="检测到的框是文本的概率，值越大，框中是文本的概率就越大。存在漏检时，调低该值。取值范围：[0, 1.0]，默认值为0.5",
+        )
+        unclip_ratio = gr.Slider(
+            label="unclip_ratio",
+            minimum=1.5,
+            maximum=2.0,
+            value=1.6,
+            step=0.1,
+            info="控制文本检测框的大小，值越大，检测框整体越大。在出现框截断文字的情况，调大该值。取值范围：[1.5, 2.0]，默认值为1.6",
+        )
+
+    img_input = gr.Image(label="Upload or Select Image", sources="upload")
+    run_btn = gr.Button("Run")
+
+    run_btn.click(test, inputs=img_input, outputs=gr.Image())
+
+    examples = gr.Examples(
+        examples=example_images,
+        examples_per_page=len(example_images),
+        inputs=img_input,
+        fn=lambda x: x,  # 简单返回图片路径
+        outputs=img_input,
+        cache_examples=False,
+    )
+
 
 if __name__ == "__main__":
     demo.launch(debug=True)