File size: 2,319 Bytes
3cd9d06 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import numpy as np
import torch
import gradio as gr
from libs import *
with gr.Blocks(title="Signal Similarity Live Demo") as demo:
default_fig, default_result = demo_fn()
with gr.Row():
signal_images = gr.Plot(default_fig,label="Visualize Signals")
with gr.Row():
singal_results = gr.Text(default_result, interactive=False)
with gr.Row():
with gr.Column():
gr.Markdown("### Ground Truth Control")
gt_button = gr.Radio(["Sin", "Cos", "Random"], label="Signal Type", value="Sin")
gt_scale_slider = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Scaling Factor", value=1)
gt_shift_slider = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Shifting Factor", value=0)
with gr.Column():
gr.Markdown("### Pred Signal Control")
signal_button = gr.Radio(["Sin", "Cos", "Random"], label="Signal Type", value="Sin")
pred_scale_slider = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Scaling Factor", value=1)
pred_shift_slider = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Shifting Factor", value=0)
gt_button.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
signal_button.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
gt_scale_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
gt_shift_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
pred_scale_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
pred_shift_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
demo.launch(share=True) |