| |
| import gradio as gr |
| import json |
| import logging |
| from PredictService import PredictService |
|
|
| log_format = "[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s" |
| logging.basicConfig(level=logging.INFO, format=log_format) |
| logger = logging.getLogger() |
|
|
| svc = PredictService() |
|
|
|
|
| def api_classification(url): |
| url_to_use = url |
| if url_to_use == "exemple": |
| url_to_use = "https://images.pexels.com/photos/326900/pexels-photo-326900.jpeg?cs=srgb&dl=pexels-pixabay-326900.jpg&fm=jpg" |
| predictions = svc.predict(url_to_use) |
| return json.dumps(predictions), url_to_use |
|
|
|
|
| with gr.Blocks() as app: |
| with gr.Tab("BioCLIP API"): |
| with gr.Row(): |
| with gr.Column(): |
| |
| gr.HTML(value=""" |
| This <a href="https://huggingface.co/spaces/3oly/grBird">Gradio BioCLIP API</a> is a <a href="https://github.com/Imageomics/bioclip">BioCLIP</a> based prediction.<br/><br/> |
| <small>origin is a <a href="https://huggingface.co/spaces/imageomics/bioclip-demo">BioCLIP DEMO</a> |
| <a href="https://huggingface.co/spaces/imageomics/bioclip-demo/discussions/7">discussion</a> |
| following <a href="https://github.com/boly38/pyBird">pyBird</a> MVP.<br/> |
| This endpoint is used by <a href="https://bsky.app/profile/botensky.bsky.social">@botEnSky</a> BlueSky bot.</small> |
| <br/> |
| <h2>How to use Gradio UI ?</h2> |
| <ul> |
| <li>You must input a <b>public url of an image</b> and submit by clicking on "predict".</li> |
| <li>You will get TreeOfLife predictions as result.</li> |
| </ul> |
| <h2>How to use Gradio API (Node.js, Python, Bash Curl) ?</h2> |
| <ul> |
| <li>You must follow the demo footer link called "Use via API". There is some example provided |
| (+ <a href="https://github.com/boly38/bioclipBootstrap?tab=readme-ov-file#content-for-3olygrbird">github bootstrap</a>).</li> |
| </ul> |
| <h2>Credits</h2> |
| <ul> |
| <li>This API endpoint is (only) an augmented fork of the <a href="https://github.com/Imageomics/bioclip">BioCLIP</a> : a wrapper that provide another form of API call.</li> |
| <li>Research context, model, and contact are available at <a href="https://github.com/Imageomics/bioclip">Imageomics/bioclip</a>.</li> |
| </ul> |
| """, show_label=False) |
| api_input = gr.Textbox( |
| lines=1, |
| label="Input a public image url", |
| show_label=False, |
| info="Add image url here.", |
| value="https://natureconservancy-h.assetsadobe.com/is/image/content/dam/tnc/nature/en/photos/d/o/Downy-woodpecker-Matt-Williams.jpg?crop=0%2C39%2C3097%2C2322&wid=820&hei=615&scl=3.776829268292683" |
| ) |
| api_classification_btn = gr.Button("predict", variant="primary") |
| api_classification_output_gallery = gr.Image(label="Input image used") |
| with gr.Column(): |
| api_classification_output_json = gr.JSON(label="This is classification result") |
|
|
| api_classification_btn.click( |
| fn=api_classification, |
| inputs=[api_input], |
| outputs=[api_classification_output_json, api_classification_output_gallery], |
| ) |
| app.queue(max_size=20) |
| app.launch() |
|
|
