dd

README.md

@@ -1,76 +1 @@
-import altair
-
-import gradio as gr
-from math import sqrt
-import matplotlib
-
-matplotlib.use("Agg")
-
-import matplotlib.pyplot as plt
-import numpy as np
-import plotly.express as px
-import pandas as pd
-
-
-def outbreak(plot_type, 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)
-    pop_count = {"USA": 350, "Canada": 40, "Mexico": 300, "UK": 120}
-    if social_distancing:
-        r = sqrt(r)
-    df = pd.DataFrame({"day": x})
-    for country in countries:
-        df[country] = x ** (r) * (pop_count[country] + 1)
-
-    if plot_type == "Matplotlib":
-        fig = plt.figure()
-        plt.plot(df["day"], df[countries].to_numpy())
-        plt.title("Outbreak in " + month)
-        plt.ylabel("Cases")
-        plt.xlabel("Days since Day 0")
-        plt.legend(countries)
-        return fig
-    elif plot_type == "Plotly":
-        fig = px.line(df, x="day", y=countries)
-        fig.update_layout(
-            title="Outbreak in " + month,
-            xaxis_title="Cases",
-            yaxis_title="Days Since Day 0",
-        )
-        return fig
-    elif plot_type == "Altair":
-        df = df.melt(id_vars="day").rename(columns={"variable": "country"})
-        fig = altair.Chart(df).mark_line().encode(x="day", y='value', color='country')
-        return fig
-    else:
-        raise ValueError("A plot type must be selected")
-
-
-inputs = [
-    gr.Dropdown(["Matplotlib", "Plotly", "Altair"], label="Plot Type"),
-    gr.Slider(1, 4, 3.2, label="R"),
-    gr.Dropdown(["January", "February", "March", "April", "May"], label="Month"),
-    gr.CheckboxGroup(
-        ["USA", "Canada", "Mexico", "UK"], label="Countries", value=["USA", "Canada"]
-    ),
-    gr.Checkbox(label="Social Distancing?"),
-]
-outputs = gr.Plot()
-
-demo = gr.Interface(
-    fn=outbreak,
-    inputs=inputs,
-    outputs=outputs,
-    examples=[
-        ["Matplotlib", 2, "March", ["Mexico", "UK"], True],
-        ["Altair", 2, "March", ["Mexico", "Canada"], True],
-        ["Plotly", 3.6, "February", ["Canada", "Mexico", "UK"], False],
-    ],
-    cache_examples=True,
-)
-
-if __name__ == "__main__":
-    demo.launch()
+Machine model