Update app.py

app.py

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