Update app.py

app.py

@@ -80,6 +80,9 @@ def process_image(model, tokenizer, transform, device, check_type, image, text):
             text_embeds = model.tok_embeddings(input_ids)
         
         vit_embeds, size1 = model.forward_tokenocr(pixel_values.to(torch.bfloat16).to(device))
+        print("vit_embeds",vit_embeds)
+        print("target_ratio",target_ratio)
+        print("check_type",check_type)
         vit_embeds, size2 = post_process(vit_embeds, target_ratio, check_type)
         
         # 计算相似度