|
|
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) |
|
|
|
