moved images to data folder, cleaned up code

.htaccess

@@ -1,3 +0,0 @@
-AddType video/ogg .ogv
-AddType video/mp4 .mp4
-AddType video/webm .webm

app.py

@@ -30,10 +30,10 @@ def get_image_data(inp_image):
     return image, model_and_data
 
 
-def segmentation_generation(image_name, c_width, n_top, n_gif_imgs):
-    print("Inputs Received:", image_name, c_width, n_top, n_gif_imgs)    
+def segmentation_generation(input_image, c_width, n_top, n_gif_imgs):
+    print("Inputs Received:", input_image, c_width, n_top, n_gif_imgs)    
 
-    image, model_and_data = get_image_data(image_name)
+    image, model_and_data = get_image_data(input_image)
     
     # Unpack datax
     xtest = model_and_data["xtest"]
@@ -76,7 +76,7 @@ def segmentation_generation(image_name, c_width, n_top, n_gif_imgs):
                             l2=False)
 
     # Create the gif of the explanation
-    return create_gif(rout['blr'], image_name, segments, instance, prediction[0], n_gif_imgs, n_top)
+    return create_gif(rout['blr'], input_image, segments, instance, prediction[0], n_gif_imgs, n_top)
 
 if __name__ == "__main__":
     inp = gr.inputs.Image(label="Input Image (Or select an example)", type="pil")
@@ -91,10 +91,10 @@ if __name__ == "__main__":
             gr.inputs.Slider(minimum=10, maximum=100, step=1, default=30, label="n_gif_images", optional=False), 
         ], 
         outputs=out, 
-        examples=[["./diego.png", 0.01, 7, 50], 
-                  ["./french_bulldog.jpg", 0.01, 5, 50],
-                  ["./pepper.jpeg", 0.01, 5, 50], 
-                  ["./bird.jpg", 0.01, 5, 50], 
-                  ["./hockey.jpg", 0.01, 5, 50]]
+        examples=[["./data/diego.png", 0.01, 7, 50], 
+                  ["./data/french_bulldog.jpg", 0.01, 5, 50],
+                  ["./data/pepper.jpeg", 0.01, 5, 50], 
+                  ["./data/bird.jpg", 0.01, 5, 50], 
+                  ["./data/hockey.jpg", 0.01, 5, 50]]
     )
     iface.launch(enable_queue=True)

image_posterior.py

@@ -82,17 +82,17 @@ def create_gif(explanation_blr, img_name, segments, image, prediction, n_images=
     
         # Save to gif
         # https://stackoverflow.com/questions/61716066/creating-an-animation-out-of-matplotlib-pngs
-        print(f"Saving gif to {img_name}_explanation.gif")
+        print(f"Saving gif to {str(prediction)}_explanation.gif")
 
-        if(os.path.exists(f'{img_name}_explanation.gif')):
-            os.remove(f'{img_name}_explanation.gif')
+        if(os.path.exists(f'{str(prediction)}_explanation.gif')):
+            os.remove(f'{str(prediction)}_explanation.gif')
 
         ims = [imageio.imread(f) for f in paths]
-        imageio.mimwrite(f'{img_name}_explanation.gif', ims)
+        imageio.mimwrite(f'{str(prediction)}_explanation.gif', ims)
 
     html = (
         "<div>"
-        + f"<img  src='file/{img_name}_explanation.gif' alt='explanation gif'/>"
+        + f"<img  src='file/{str(prediction)}_explanation.gif' alt='explanation gif'/>"
         + "</div>"
     )