Upload with huggingface_hub

DESCRIPTION.md

@@ -0,0 +1 @@
+Generate a plot based on 5 inputs.

README.md

@@ -1,12 +1,11 @@
+
 ---
-title: Outbreak Forecast Main
-emoji: 📊
-colorFrom: yellow
-colorTo: gray
+title: outbreak_forecast_main 
+emoji: 🔥
+colorFrom: indigo
+colorTo: indigo
 sdk: gradio
 sdk_version: 3.6
-app_file: app.py
+app_file: run.py
 pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

requirements.txt

@@ -0,0 +1,4 @@
+numpy
+matplotlib
+bokeh
+plotlyhttps://gradio-main-build.s3.amazonaws.com/c3bec6153737855510542e8154391f328ac72606/gradio-3.6-py3-none-any.whl

run.py

@@ -0,0 +1,69 @@
+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
+    else:
+        raise ValueError("A plot type must be selected")
+
+
+inputs = [
+    gr.Dropdown(["Matplotlib", "Plotly"], 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],
+        ["Plotly", 3.6, "February", ["Canada", "Mexico", "UK"], False],
+    ],
+    cache_examples=True,
+)
+
+if __name__ == "__main__":
+    demo.launch()