starting to implement image_posterior model code

app.py

@@ -5,6 +5,7 @@ import argparse
 import os
 from os.path import exists, dirname
 import sys
+import flask
 
 parent_dir = dirname(os.path.abspath(os.getcwd()))
 sys.path.append(parent_dir)
@@ -78,11 +79,8 @@ def image_mod(image):
 
 
 if __name__ == "__main__":
-    args = parser.parse_args()
-    # main(args)
-
     inp = gr.inputs.Image(label="Input Image", type="pil")
-    out = gr.outputs.Image()
+    out = gr.outputs.Video(label="Output Image")
 
     iface = gr.Interface(segmentation_generation, inputs=inp, outputs=out, examples=[["./imagenet_diego.png"]])
     iface.launch()

image_posterior.py

@@ -18,7 +18,7 @@ import numpy as np
 from skimage.segmentation import mark_boundaries
 import tempfile
 from tqdm import tqdm
-
+import ffmpeg
 import lime.lime_tabular as baseline_lime_tabular
 import shap
 
@@ -67,7 +67,7 @@ def create_gif(explanation_blr, segments, image, n_images=20, n_max=5):
 
         # Save to gif
         # https://stackoverflow.com/questions/61716066/creating-an-animation-out-of-matplotlib-pngs
-        print(f"Saving gif to {save_loc}")
+        # print(f"Saving gif to {save_loc}")
         ims = [imageio.imread(f) for f in paths]
         return imageio.mimwrite(imageio.RETURN_BYTES, ims)