add spaces

app.py

@@ -40,7 +40,7 @@ def check_move(trajectory, video_length=16):
 
     return True
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=600)
 def infer(*user_args):
     prompt_in = user_args[0]
     target_indices = user_args[1]
@@ -86,7 +86,10 @@ def infer(*user_args):
         w_relative = i[2] * w_remain
         input_traj.append([i[0], h_relative, h_relative+height_ratio, w_relative, w_relative+width_ratio])
 
-    indices_list = target_indices.split(',')
+    if len(target_indices) < 1:
+        indices_list = [1, 2]
+    else:
+        indices_list = target_indices.split(',')
     idx_list = []
     for i in indices_list:
         idx_list.append(int(i))
@@ -575,7 +578,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Group():
             with gr.Row():
                 prompt_in = gr.Textbox(label="Prompt", placeholder="A corgi running on the grassland on the grassland.", scale = 3)
-                target_indices = gr.Textbox(label="Target Indices", placeholder="1,2", scale = 1)
+                target_indices = gr.Textbox(label="Target Indices (1 for the first word, necessary!)", placeholder="1,2", scale = 1)
 
             with gr.Row():
                 radio_mode = gr.Radio(label='Trajectory Mode', choices = ['demo', 'diy', 'ori'], scale = 1)