Update app.py

app.py

@@ -166,6 +166,11 @@ def main_function(plan_name, uploaded_file, interference = True, aps_int = 0, ap
     # Load plan
     if plan_name == "Upload your own image" and uploaded_file is not None:
         plan_image = np.array(Image.open(uploaded_file.name))/255
+
+        dimensiones = plan_image.shape
+        if dimensiones[0] != 256 or dimensiones[1] != 256 or dimensiones[3] != 3:
+            raise ValueError("Error in image size. Must [256,256,3].")
+        
         plan_grayscale = plan_image[:, :, 0]
         plan_in = 1 - plan_grayscale