Update app.py

app.py

@@ -454,7 +454,7 @@ def generate_video(image, prompt, max_tokens):
     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)
     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=test_image,target_aspect_ratio=target_aspect_ratio)
+                            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