pacman2223 commited on
Commit
2d248e3
·
verified ·
1 Parent(s): a891068

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def demo_process(img, question):
19
  processor = AutoProcessor.from_pretrained(model_checkpoint)
20
  model = AutoModelForDocumentQuestionAnswering.from_pretrained(model_checkpoint)
21
  with torch.no_grad():
22
- encoding = processor(img.convert("RGB"), question, return_tensors="pt")
23
  outputs = model(**encoding)
24
  start_logits = outputs.start_logits
25
  end_logits = outputs.end_logits
 
19
  processor = AutoProcessor.from_pretrained(model_checkpoint)
20
  model = AutoModelForDocumentQuestionAnswering.from_pretrained(model_checkpoint)
21
  with torch.no_grad():
22
+ encoding = processor(img, question, return_tensors="pt")
23
  outputs = model(**encoding)
24
  start_logits = outputs.start_logits
25
  end_logits = outputs.end_logits