Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|