Update SIR_model.py
Browse files- SIR_model.py +3 -0
SIR_model.py
CHANGED
|
@@ -10,6 +10,9 @@ import matplotlib.pyplot as plt
|
|
| 10 |
import os
|
| 11 |
import sys
|
| 12 |
import streamlit as st
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
def main(total_population, initial_infected, initial_recovered, beta, gamma, mode=None):
|
| 15 |
# Initialize the Model
|
|
|
|
| 10 |
import os
|
| 11 |
import sys
|
| 12 |
import streamlit as st
|
| 13 |
+
from matplotlib.animation import FuncAnimation
|
| 14 |
+
|
| 15 |
+
# Rest of the code remains the same
|
| 16 |
|
| 17 |
def main(total_population, initial_infected, initial_recovered, beta, gamma, mode=None):
|
| 18 |
# Initialize the Model
|