credent007 commited on
Commit
54b020c
·
verified ·
1 Parent(s): b17ed36

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +8 -1
inference.py CHANGED
@@ -4,7 +4,14 @@ from processor_utils import load_input
4
  from prompt import get_prompt
5
  import json
6
  def process_document(file_path):
7
- image = load_input(file_path)
 
 
 
 
 
 
 
8
 
9
  messages = [
10
  {
 
4
  from prompt import get_prompt
5
  import json
6
  def process_document(file_path):
7
+ images = load_input(file_path)
8
+ image = images[0]
9
+ print("Checking input type and no of pages in pdf")
10
+ print(type(image))
11
+ print(type(images))
12
+ print(len(images))
13
+
14
+
15
 
16
  messages = [
17
  {