Hugging Face's logo

Spaces:
andgrt
/
doc-qa-demo-gradio
Runtime error

App Files Community
andgrt commited on
Commit
15151eb
·
1 Parent(s): 0618e26

fix: correct processor

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -7,16 +7,13 @@ from transformers import (
7
  )
8
  import torch
9
 
10
-
11
  tokenizer_ru2en = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
12
  model_ru2en = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
13
 
14
  tokenizer_en2ru = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ru")
15
  model_en2ru = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-ru")
16
 
17
- git_processor_base = AutoProcessor.from_pretrained(
18
- "andgrt/layoutlmv2-base-uncased_finetuned_docvqa"
19
- )
20
 
21
  git_model_base = AutoModelForDocumentQuestionAnswering.from_pretrained(
22
  "andgrt/layoutlmv2-base-uncased_finetuned_docvqa"
 
7
  )
8
  import torch
9
 
 
10
  tokenizer_ru2en = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
11
  model_ru2en = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
12
 
13
  tokenizer_en2ru = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ru")
14
  model_en2ru = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-ru")
15
 
16
+ git_processor_base = AutoProcessor.from_pretrained("microsoft/layoutlmv2-base-uncased")
 
 
17
 
18
  git_model_base = AutoModelForDocumentQuestionAnswering.from_pretrained(
19
  "andgrt/layoutlmv2-base-uncased_finetuned_docvqa"