Update app.py

app.py

@@ -24,7 +24,7 @@ def deriv(x, t, beta, gamma):
     return [dsdt, didt, drdt]
 
 #plot model
-def plotdata(t, s, i,r, e=None):
+def plotdata(t, s, i,r,R0, e=None):
     # plot the data
     fig = plt.figure(figsize=(12,6))
     ax = [fig.add_subplot(221, axisbelow=True), 
@@ -82,7 +82,7 @@ def SIR(country,t_infective):
   s, i, r = soln.T
   e = None
 
-  plotdata(t, s, i,r)
+  plotdata(t, s, i,r,R0)
 
   return R0,t_infective,gamma,beta