edit Image

app.py

@@ -43,7 +43,7 @@ def cohort_predict(start_date, cohort_start, product):
     fig = plot_example_from_case(hist_seq, new_seq, 25, product)
         
               
-    return fig
+    return Image.frombytes('RGB', fig.canvas.get_width_height(),fig.canvas.tostring_rgb())
         
 demo = gr.Interface(
     fn=cohort_predict,