Update app.py

app.py

@@ -396,7 +396,7 @@ def generate_video(image, prompt, max_tokens):
     print(image)
     pixel_values, target_aspect_ratio = load_image(image, max_num=6)
     pixel_values = pixel_values.to(torch.bfloat16).cuda()
-    generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 3, repetition_penalty=2.5)
+    generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 1, 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)