TongkunGuan commited on
Commit
f171faa
·
verified ·
1 Parent(s): 1afeeab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -81,9 +81,9 @@ def process_image(model, tokenizer, transform, device, check_type, image, text):
81
 
82
  vit_embeds, size1 = model.forward_tokenocr(pixel_values.to(torch.bfloat16).to(device))
83
 
84
- print("vit_embeds",vit_embeds)
85
- print("vit_embeds,shape",vit_embeds.shape)
86
- print("target_ratio",target_ratio)
87
  print("check_type",check_type)
88
  vit_embeds, size2 = post_process(vit_embeds, target_ratio, check_type)
89
 
 
81
 
82
  vit_embeds, size1 = model.forward_tokenocr(pixel_values.to(torch.bfloat16).to(device))
83
 
84
+ # print("vit_embeds",vit_embeds)
85
+ # print("vit_embeds,shape",vit_embeds.shape)
86
+ # print("target_ratio",target_ratio)
87
  print("check_type",check_type)
88
  vit_embeds, size2 = post_process(vit_embeds, target_ratio, check_type)
89