Update app.py

app.py

@@ -170,9 +170,9 @@ def main_function(plan_name, uploaded_file, interference = True, aps_int = 0, ap
         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:
+        if len(dimensiones) > 2:
             # raise gradio.Error("Error in dimensions of the uploaded image. Must [256,256,3] 💣🎆.", duration=5)
-            raise ValueError("Error in image size. Must [256,256,3].")
+            raise ValueError("Error in image size. Must [256,256].")
         
         plan_grayscale = plan_image[:, :, 0]
         plan_in = 1 - plan_grayscale