mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-15 02:11:15 +08:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import gradio as gr
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
from math import sqrt
|
|
|
|
def outbreak(r, month, countries, social_distancing):
|
|
months = ["January", "February", "March", "April", "May"]
|
|
m = months.index(month)
|
|
start_day = 30 * m
|
|
final_day = 30 * (m + 1)
|
|
x = np.arange(start_day, final_day+1)
|
|
day_count = x.shape[0]
|
|
pop_count = {"USA": 350, "Canada": 40, "Mexico": 300, "UK": 120}
|
|
r = sqrt(r)
|
|
if social_distancing:
|
|
r = sqrt(r)
|
|
for i, country in enumerate(countries):
|
|
series = x ** (r) * (i + 1)
|
|
plt.plot(x, series)
|
|
plt.title("Outbreak in " + month)
|
|
plt.ylabel("Cases")
|
|
plt.xlabel("Days since Day 0")
|
|
plt.legend(countries)
|
|
return plt
|
|
|
|
iface = gr.Interface(outbreak,
|
|
[
|
|
gr.inputs.Slider(1, 4, default=3.2, label="R"),
|
|
gr.inputs.Dropdown(["January", "February", "March", "April", "May"], label="Month"),
|
|
gr.inputs.CheckboxGroup(["USA", "Canada", "Mexico", "UK"], label="Countries"),
|
|
gr.inputs.Checkbox(label="Social Distancing?"),
|
|
],
|
|
"plot")
|
|
if __name__ == "__main__":
|
|
iface.launch()
|