| import gradio as gr |
| from PIL import Image |
| import hopsworks |
|
|
| |
| from hopsworks.core.dataset_api import DatasetApi |
|
|
| |
| hopsworks_images_location = "Resources/images" |
| hopsworks_images = { |
| "latest_bridge": |
| { |
| "name": "latest_bridge/latest_bridge.png", |
| "local_path": "" |
| }, |
| "actual_bridge": |
| { |
| "name": "latest_bridge/actual_bridge.png", |
| "local_path": "" |
| }, |
| "df_recent": |
| { |
| "name": "latest_bridge/df_recent.png", |
| "local_path": "" |
| } |
| } |
|
|
| print("Logging in to Hopsworks...") |
| project = hopsworks.login() |
|
|
| print("Getting feature store...") |
| fs = project.get_feature_store() |
|
|
| print("Get database handler from Hopsworks...") |
| dataset_api: DatasetApi = project.get_dataset_api() |
|
|
| for image in hopsworks_images: |
| print(f"Downloading {hopsworks_images[image]['name']} from Hopsworks...") |
| hopsworks_images[image]['local_path'] = \ |
| dataset_api.download(f"{hopsworks_images_location}/{hopsworks_images[image]['name']}", overwrite=True) |
| print(f"Saved in: {hopsworks_images[image]['local_path']}") |
|
|
| print("Configuring gradio...") |
| with gr.Blocks() as demo: |
| with gr.Row(): |
| with gr.Column(): |
| label1 = gr.Label("Last Predicted Bridge Status") |
| image1 = gr.Image(hopsworks_images['latest_bridge']['local_path'], |
| elem_id="predicted-bridge-img", type="pil") |
| with gr.Column(): |
| label2 = gr.Label("Last Actual Bridge Status") |
| image2 = gr.Image(hopsworks_images['actual_bridge']['local_path'], |
| elem_id="actual-bridge-img", type="pil") |
| with gr.Row(): |
| with gr.Column(): |
| label3 = gr.Label("Recent Bridge Status Prediction History") |
| image3 = gr.Image(hopsworks_images['df_recent']['local_path'], |
| elem_id="recent-bridge-predictions", type="pil") |
| with gr.Column(scale=2): |
| image_credits = gr.Label( |
| "Image Credits: Open bridge image from " |
| "[Wikimedia Commons](https://commons.wikimedia.org/wiki/File:M%C3%A4larbron_juni_2017f.jpg), " |
| "Closed bridge image from " |
| "[Wikimedia Commons](https://commons.wikimedia.org/wiki/File:M%C3%A4larbron_juni_2017e.jpg)") |
|
|
| print("Launching gradio...") |
| demo.launch(debug=True) |
|
|
