test

app.py

@@ -321,15 +321,18 @@ def create_demo(model):
                     images[i].append(inputs[i * max_grid_w + j])
                     if i != model.grid_h - 1:
                         if inputs[i * max_grid_w + j] is None:
-                            raise gr.Error('Please upload in-context examples.')
+                            raise gr.Error('Please upload in-context examples. Possible that the task examples have not finished loading yet.')
             seed, cfg, steps, upsampling_steps, upsampling_noise, layout_text, task_text, content_text = inputs[-8:]
-                
-            results = generate(
-                images, 
-                [layout_text, task_text, content_text], 
-                seed=seed, cfg=cfg, steps=steps, 
-                upsampling_steps=upsampling_steps, upsampling_noise=upsampling_noise
-            )
+            
+            try:
+                results = generate(
+                    images, 
+                    [layout_text, task_text, content_text], 
+                    seed=seed, cfg=cfg, steps=steps, 
+                    upsampling_steps=upsampling_steps, upsampling_noise=upsampling_noise
+                )
+            except Exception as e:
+                raise gr.Error('Process error. Possible that the task examples have not finished loading yet. Error: ' + e)
 
             output = gr.update(
                 elem_id='output_gallery', 
@@ -471,7 +474,7 @@ def create_demo(model):
 
 
 # @spaces.GPU(duration=120)
-@spaces.GPU()
+@spaces.GPU(5)
 def generate(
     images, 
     prompts,