khang119966 commited on
Commit
6bde22d
·
verified ·
1 Parent(s): cb3ffc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ tokenizer = AutoTokenizer.from_pretrained("khang119966/Vintern-1B-v3_5-explainab
144
 
145
  @spaces.GPU
146
  def generate_video(image, prompt, max_tokens):
147
- pixel_values, target_aspect_ratio = load_image(test_image, max_num=6).to(torch.bfloat16).cuda()
148
  generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 3, repetition_penalty=2.5)
149
  response, query = model.chat(tokenizer, pixel_values, '<image>\n'+prompt, generation_config, return_history=False, \
150
  attention_visualize=True,last_visualize_layers=7,raw_image_path=test_image,target_aspect_ratio=target_aspect_ratio)
 
144
 
145
  @spaces.GPU
146
  def generate_video(image, prompt, max_tokens):
147
+ pixel_values, target_aspect_ratio = load_image(image, max_num=6).to(torch.bfloat16).cuda()
148
  generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 3, repetition_penalty=2.5)
149
  response, query = model.chat(tokenizer, pixel_values, '<image>\n'+prompt, generation_config, return_history=False, \
150
  attention_visualize=True,last_visualize_layers=7,raw_image_path=test_image,target_aspect_ratio=target_aspect_ratio)