Spaces:
Running
Running
| import os | |
| import requests | |
| import gradio as gr | |
| url = os.environ["URL_NODE"] | |
| def detect_image(image): | |
| print("image: ", image) | |
| files = {"picture": open(image, "rb")} | |
| resp = requests.post(url, | |
| files=files, | |
| verify=False) | |
| resp = resp.json() | |
| gen_url = resp["data"]["answer"] | |
| return gen_url | |
| def read_content(file_path): | |
| with open(file_path, 'r', encoding='utf-8') as f: | |
| content = f.read() | |
| return content | |
| example_images = [ | |
| os.path.join(os.path.dirname(__file__), "examples/00.jpg"), | |
| os.path.join(os.path.dirname(__file__), "examples/01.jpg"), | |
| os.path.join(os.path.dirname(__file__), "examples/02.jpg"), | |
| os.path.join(os.path.dirname(__file__), "examples/03.jpg"), | |
| os.path.join(os.path.dirname(__file__), "examples/04.jpg"), | |
| os.path.join(os.path.dirname(__file__), "examples/05.jpg"), | |
| os.path.join(os.path.dirname(__file__), "examples/06.png") | |
| ] | |
| default_image = example_images[0] | |
| css = """ | |
| .gradio-container {background-image: url('file=./background.jpg'); background-size:cover; background-repeat: no-repeat;} | |
| """ | |
| # warm up | |
| # detect_image() | |
| with gr.Blocks(css=css) as demo: | |
| gr.HTML(read_content("./header.html")) | |
| gr.Markdown("# MindSpore Wuhan.LuoJiaNET") | |
| gr.Markdown( | |
| "`Wuhan.LuoJiaNET` is the first domestic autonomous and controllable machine learning framework for remote sensing in the field of remote sensing," | |
| " jointly developed by` Wuhan University` and `Huawei's Ascend AI team`, which has the characteristics of large image size," | |
| " multiple data channels, and large scale variation of remote sensing data." | |
| " It is compatible with existing deep learning frameworks and provides a user-friendly," | |
| " drag-and-drop interactive network structure to build an interface." | |
| " It can shield the differences between different hardware devices and manage a diversified remote sensing image sample library," | |
| " LuoJiaSET, to achieve efficient storage and management of remote multi-source sensing image samples." | |
| ) | |
| with gr.Tab("目标识别 (Object Detection)"): | |
| with gr.Row(): | |
| image_input = gr.Image( | |
| type="filepath", | |
| value=default_image | |
| ) | |
| image_output = gr.Image( | |
| type="filepath", | |
| interactive=False | |
| ) | |
| gr.Examples( | |
| examples=example_images, | |
| inputs=image_input, | |
| ) | |
| image_button = gr.Button("Detect") | |
| with gr.Accordion("Open for More!"): | |
| gr.Markdown( | |
| "- If you want to know more about the foundation models of MindSpore, please visit " | |
| "[The Foundation Models Platform for Mindspore](https://xihe.mindspore.cn/)" | |
| ) | |
| gr.Markdown( | |
| "- If you want to know more about Wuhan.LuoJiaNET, please visit " | |
| "[Wuhan.LuoJiaNET](https://github.com/WHULuoJiaTeam/luojianet)") | |
| gr.Markdown( | |
| "- Try [Wukong-LuojiaNET model on the Foundation Models Platform for Mindspore]" | |
| "(https://xihe.mindspore.cn/modelzoo/luojia)") | |
| image_button.click(detect_image, | |
| inputs=[image_input], | |
| outputs=[image_output]) | |
| demo.queue(concurrency_count=5) | |
| demo.launch(enable_queue=True) |