Santiago Roman commited on
Commit
0b6fd25
·
1 Parent(s): 0e2f3d4
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -41,6 +41,7 @@ def cohort_predict(start_date, cohort_start, product):
41
  hist_seq = get_sequence(data, cohort_start, product)
42
 
43
  fig = plot_example_from_case(hist_seq, new_seq, 25, product)
 
44
 
45
  arr = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
46
  arr = arr.reshape(fig.canvas.get_width_height()[::-1] + (3,))
 
41
  hist_seq = get_sequence(data, cohort_start, product)
42
 
43
  fig = plot_example_from_case(hist_seq, new_seq, 25, product)
44
+ fig.canvas.draw()
45
 
46
  arr = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
47
  arr = arr.reshape(fig.canvas.get_width_height()[::-1] + (3,))