Update app.py

app.py

@@ -92,7 +92,7 @@ def SIR(country,t_infective):
   r_initial = 0.00
   s_initial = 1 - i_initial - r_initial
   
-  t = np.linspace(0, 100, 1000)
+  t = np.linspace(0, 1000, 1000)
   x_initial = s_initial, i_initial, r_initial
   soln = odeint(deriv, x_initial, t, args=(beta, gamma))
   s, i, r = soln.T
@@ -107,13 +107,13 @@ def main():
     with st.form("questionaire"):
         country = st.selectbox("Country",data['location'].unique())# user's input
         recovery = st.slider("How long Monkeypox recover?", 21, 31, 21)# user's input
-        country = code[country][0]
+        country_code = code[country][0]
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Show Graph")
         if clicked:
         
             # Show SIR
-            SIR_param = SIR(country,recovery)
+            SIR_param = SIR(country_code,recovery)
             
             st.pyplot(SIR_param[-1])
             st.success("SIR model parameters for "+str(country)+" is")