fix image

app.py

@@ -41,9 +41,11 @@ def cohort_predict(start_date, cohort_start, product):
     hist_seq = get_sequence(data, cohort_start, product)
 
     fig = plot_example_from_case(hist_seq, new_seq, 25, product)
-        
-              
-    return Image.frombytes('RGB', fig.canvas.get_width_height(),fig.canvas.tostring_rgb())
+
+    arr = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+    arr = arr.reshape(fig.canvas.get_width_height()[::-1] + (3,))    
+    
+    return Image.fromarray(arr)
         
 demo = gr.Interface(
     fn=cohort_predict,