removing hardcoded aspects

app.py

@@ -1,4 +1,6 @@
 from cgitb import enable
+from pyexpat import model
+from statistics import mode
 import numpy as np
 import gradio as gr
 """An example of generating a gif explanation for an image of my dog."""
@@ -21,6 +23,7 @@ BLENHEIM_SPANIEL_CLASS = 156
 
 def get_image_data(image_name):
     """Gets the image data and model."""
+    image, model_and_data = [None, None]
     if (image_name == "imagenet_diego.png"): 
         image = get_dataset_by_name("imagenet_diego", get_label=False)
         model_and_data = process_imagenet_get_model(image)
@@ -39,7 +42,7 @@ def segmentation_generation(image_name, c_width, n_top, n_gif_imgs):
     n_top_segs = n_top
     n_gif_images = n_gif_imgs
     image, model_and_data = get_image_data(image_name)
-
+    print("model_and_data", model_and_data)
     # Unpack datax
     xtest = model_and_data["xtest"]
     ytest = model_and_data["ytest"]