Update app.py

app.py

@@ -459,7 +459,7 @@ def generate_text_img_wrapper(args):
 def generate_hidden_img_wrapper(args):
     return render_next_token_table_image(*args)
         
-@spaces.GPU
+@spaces.GPU(duration=120)
 def generate_video(image, prompt, max_tokens):
     print(image)
     pixel_values, target_aspect_ratio = load_image(image, max_num=6)
@@ -467,7 +467,7 @@ def generate_video(image, prompt, max_tokens):
     generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 3, repetition_penalty=2.5)
     response, query = model.chat(tokenizer, pixel_values, '<image>\n'+prompt, generation_config, return_history=False, \
                             attention_visualize=True,last_visualize_layers=7,raw_image_path=image,target_aspect_ratio=target_aspect_ratio)
-    print(response)
+
     generation_output = response
     raw_image_path = image