debugging index 0 is out of bounds for axis 0 with size 0 error

app.py

@@ -30,11 +30,14 @@ BLENHEIM_SPANIEL_CLASS = 156
 def get_image_data():
     """Gets the image data and model."""
     puppy_image = get_dataset_by_name(IMAGE_NAME, get_label=False)
+    print("IMAGE RETURNED FROM GETTING DATASET:\n", model_and_data)
     model_and_data = process_imagenet_get_model(puppy_image)
+    print("MODEL RETURNED FROM PROCESSING IMAGE:\n", model_and_data)
     return puppy_image, model_and_data
 
 
 def segmentation_generation(image_name, c_width, n_top, n_gif_imgs):
+    print("GRADIO INPUTS:", image_name, c_width, n_top, n_gif_images)
     cred_width = c_width
     n_top_segs = n_top
     n_gif_images = n_gif_imgs

bayes/models.py

@@ -39,7 +39,7 @@ def get_xtrain(segs):
 
 def process_imagenet_get_model(data):
     """Gets wrapped imagenet model."""
-
+    print("DATA RECEIVED FROM APP:\n", data)
     # Get the vgg16 model, used in the experiments
     model = models.vgg16(pretrained=True)
     model.eval()