WayneLinn's picture
added load and refresh button
ccf823d
raw
history blame contribute delete
No virus
2.46 kB
import gradio as gr
from PIL import Image
import hopsworks
import os
import imageio.v3 as iio
import inspect
project = hopsworks.login(project="test42",api_key_value=os.environ.get("HOPSWORKS_API_KEYS"))
fs = project.get_feature_store()
dataset_api = project.get_dataset_api()
print(inspect.signature(dataset_api.download))
dataset_api.download("Resources/images/latest_iris.png",overwrite=True)
dataset_api.download("Resources/images/actual_iris.png",overwrite=True)
dataset_api.download("Resources/images/df_recent.png",overwrite=True)
dataset_api.download("Resources/images/confusion_matrix.png",overwrite=True)
x=0
def update():
dataset_api.download("Resources/images/latest_iris.png",overwrite=True)
dataset_api.download("Resources/images/actual_iris.png",overwrite=True)
dataset_api.download("Resources/images/df_recent.png",overwrite=True)
dataset_api.download("Resources/images/confusion_matrix.png",overwrite=True)
def update_latest_iris_img():
im = iio.imread('latest_iris.png')
return im
def update_actual_iris_img():
im = iio.imread('actual_iris.png')
return im
def update_df_recent_img():
im = iio.imread('df_recent.png')
return im
def update_confusion_matrix_img():
im = iio.imread('confusion_matrix.png')
return im
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
load=gr.Button("Load")
load.click(fn=update)
with gr.Column():
refresh=gr.Button("Refresh")
with gr.Row():
with gr.Column():
gr.Label("Today's Predicted Image")
input_img = gr.Image("latest_iris.png", elem_id="predicted-img")
refresh.click(update_latest_iris_img,outputs=input_img)
with gr.Column():
gr.Label("Today's Actual Image")
input_img = gr.Image("actual_iris.png", elem_id="actual-img")
refresh.click(update_actual_iris_img,outputs=input_img)
with gr.Row():
with gr.Column():
gr.Label("Recent Prediction History")
input_img = gr.Image("df_recent.png", elem_id="recent-predictions")
refresh.click(update_df_recent_img,outputs=input_img)
with gr.Column():
gr.Label("Confusion Maxtrix with Historical Prediction Performance")
input_img = gr.Image("confusion_matrix.png", elem_id="confusion-matrix")
refresh.click(update_confusion_matrix_img,outputs=input_img)
demo.launch()