rosemariafontana commited on
Commit
3152d9a
Β·
verified Β·
1 Parent(s): eaebdde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,9 +2,9 @@ import gradio as gr
2
  import pandas as pd
3
 
4
  from PIL import Image
5
- from transformers import LayoutLMv2Processor, LayoutLMv3ForQuestionAnswering
6
 
7
- processor = LayoutLMv2Processor.from_pretrained("microsoft/layoutlmv3-base")
8
  model = LayoutLMv3ForQuestionAnswering.from_pretrained("microsoft/layoutlmv3-base")
9
 
10
  def process_question(question, document):
@@ -42,7 +42,7 @@ def parse_ticket_image(image, question):
42
  # Processing the image
43
  if image:
44
  try:
45
- document = Image.open(image).convert("RGB")
46
  except Exception as e:
47
  traceback.print_exc()
48
  error = str(e)
 
2
  import pandas as pd
3
 
4
  from PIL import Image
5
+ from transformers import LayoutLMv3Processor, LayoutLMv3ForQuestionAnswering
6
 
7
+ processor = LayoutLMv3Processor.from_pretrained("microsoft/layoutlmv3-base")
8
  model = LayoutLMv3ForQuestionAnswering.from_pretrained("microsoft/layoutlmv3-base")
9
 
10
  def process_question(question, document):
 
42
  # Processing the image
43
  if image:
44
  try:
45
+ document = Image.open(image.name).convert("RGB")
46
  except Exception as e:
47
  traceback.print_exc()
48
  error = str(e)