Update app.py

app.py

@@ -61,8 +61,8 @@ def plotdata(t, s, i,r,R0, e=None):
         a.legend()
 
     plt.tight_layout()
-    
-    return fig
+    st.pyplot(fig)
+     
 
 #final model
 def SIR(country,t_infective):
@@ -85,7 +85,7 @@ def SIR(country,t_infective):
   e = None
 
 
-  return plotdata(t, s, i,r,R0),R0,t_infective,gamma,beta
+  return R0,t_infective,beta,gamma
 
 def main():
     st.title("SIR Model for Monkeypox")
@@ -104,11 +104,11 @@ def main():
             # Show SIR
             SIR_param = SIR(country,recovery)
             
-            st.success(st.pyplot(plt.plot(all_location[country]['total_cases'])))
-            st.success(st.pyplot(SIR(country,recovery)[0]))
+            st.success(st.pyplot(all_location[country]['total_cases']))
+            st.success(st.pyplot(SIR(country,recovery)))
             st.success("SIR model parameters for "+str(country)+" is")
-            st.success("R0 = "+str(SIR_param[1]))
-            st.success("Beta = "+str(SIR_param[4]))
+            st.success("R0 = "+str(SIR_param[0]))
+            st.success("Beta = "+str(SIR_param[2]))
             st.success("Gamma = "+str(SIR_param[3]))
 
 # Run main()