Update SIR_model.py

SIR_model.py

@@ -11,7 +11,7 @@ import os
 import sys
 import streamlit as st
 
-def main(mode=None):
+def main(total_population, initial_infected, initial_recovered, beta, gamma, mode=None):
     # Initialize the Model
     sir_model = Model(id="SIR_Model")
 
@@ -19,12 +19,7 @@ def main(mode=None):
     sir_graph = Graph(id="SIR_Graph")
     sir_model.graphs.append(sir_graph)
 
-    # Get user inputs for model parameters
-    total_population = st.number_input("Total Population", min_value=1, value=1000, step=1)
-    initial_infected = st.number_input("Initial Infected Population", min_value=0, max_value=total_population, value=1)
-    initial_recovered = st.number_input("Initial Recovered Population", min_value=0, max_value=total_population - initial_infected, value=0)
-    beta = st.slider("Infection Rate (β)", min_value=0.0, max_value=1.0, value=0.3, step=0.01)
-    gamma = st.slider("Recovery Rate (γ)", min_value=0.0, max_value=1.0, value=0.1, step=0.01)
+
 
     # # Parameters for the model
     # total_population = 1000