credent007 commited on
Commit
89461a9
·
verified ·
1 Parent(s): 7e33445

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -47,7 +47,7 @@ def process_document(image):
47
 
48
 
49
  generated_ids = output[0][inputs.input_ids.shape[-1]:]
50
- print(f"The generated_ids of inference is {generated_ids}")
51
 
52
  # response = processor.decode( # past code
53
  # generated_ids,
@@ -62,7 +62,7 @@ def process_document(image):
62
  generated_ids,
63
  skip_special_tokens=True
64
  ).strip()
65
- print(f"The response of inference is {response}")
66
 
67
  # 🔥 FORCE JSON CLEANING
68
  start = response.find("{")
 
47
 
48
 
49
  generated_ids = output[0][inputs.input_ids.shape[-1]:]
50
+ # print(f"The generated_ids of inference is {generated_ids}")
51
 
52
  # response = processor.decode( # past code
53
  # generated_ids,
 
62
  generated_ids,
63
  skip_special_tokens=True
64
  ).strip()
65
+ # print(f"The response of inference is {response}")
66
 
67
  # 🔥 FORCE JSON CLEANING
68
  start = response.find("{")