test

app.py

@@ -312,6 +312,8 @@ def create_demo(model):
 
         def generate_image(*inputs):
             images = []
+            if grid_h != model.grid_h or grid_w != model.grid_w:
+                raise gr.Error('Please wait for the loading to complete.')
             for i in range(model.grid_h):
                 images.append([])
                 for j in range(model.grid_w):