Update app.py

app.py

@@ -102,7 +102,9 @@ def main():
 
             # Show SIR
             SIR_param = SIR(country,recovery)
-
+            
+            st.success(all_location[country]['total_cases'].plot())
+            st.success(SIR(country,recovery))
             st.success("SIR model parameters for "+str(country)+" is")
             st.success("R0 = "+str(SIR_param[0]))
             st.success("Beta = "+str(SIR_param[3]))