gradio/demo/form_graph.py

36 lines
1.2 KiB
Python
Raw Normal View History

2020-06-16 02:57:49 +08:00
import gradio as gr
import random
import matplotlib.pyplot as plt
import numpy as np
2020-07-09 06:09:52 +08:00
2020-06-16 02:57:49 +08:00
def plot_forecast(final_year, companies, noise, show_legend, point_style):
start_year = 2020
x = np.arange(start_year, final_year + 1)
year_count = x.shape[0]
plt_format = ({"cross": "X", "line": "-", "circle": "o--"})[point_style]
2020-09-01 00:54:14 +08:00
fig = plt.figure()
2020-09-08 02:40:57 +08:00
ax = fig.add_subplot(111)
2020-06-16 02:57:49 +08:00
for i, company in enumerate(companies):
series = np.arange(0, year_count, dtype=float)
series = series ** 2 * (i + 1)
series += np.random.rand(year_count) * noise
2020-09-01 00:54:14 +08:00
ax.plot(x, series, plt_format)
2020-06-16 02:57:49 +08:00
if show_legend:
plt.legend(companies)
2020-09-01 00:54:14 +08:00
return fig
2020-06-16 02:57:49 +08:00
2020-07-09 06:09:52 +08:00
2020-06-16 02:57:49 +08:00
gr.Interface(plot_forecast,
[
gr.inputs.Radio([2025, 2030, 2035, 2040],
label="Project to:"),
gr.inputs.CheckboxGroup(
["Google", "Microsoft", "Gradio"], label="Company Selection"),
gr.inputs.Slider(1, 100, label="Noise Level"),
gr.inputs.Checkbox(label="Show Legend"),
gr.inputs.Dropdown(["cross", "line", "circle"], label="Style"),
2020-06-19 04:15:27 +08:00
],
gr.outputs.Image(plot=True, label="forecast")
).launch()