andgrt commited on
Commit
fbf438c
·
1 Parent(s): 64afb51

fix: AutoModel type for docvqa

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -3,11 +3,12 @@ from transformers import (
3
  AutoTokenizer,
4
  AutoModelForSeq2SeqLM,
5
  AutoProcessor,
6
- AutoModelForCausalLM,
7
  )
8
  import torch
9
  import pyttsx3
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
 
@@ -17,7 +18,8 @@ model_en2ru = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-ru"
17
  git_processor_base = AutoProcessor.from_pretrained(
18
  "andgrt/layoutlmv2-base-uncased_finetuned_docvqa"
19
  )
20
- git_model_base = AutoModelForCausalLM.from_pretrained(
 
21
  "andgrt/layoutlmv2-base-uncased_finetuned_docvqa"
22
  )
23
 
 
3
  AutoTokenizer,
4
  AutoModelForSeq2SeqLM,
5
  AutoProcessor,
6
+ AutoModelForDocumentQuestionAnswering,
7
  )
8
  import torch
9
  import pyttsx3
10
 
11
+
12
  tokenizer_ru2en = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
13
  model_ru2en = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-ru-en")
14
 
 
18
  git_processor_base = AutoProcessor.from_pretrained(
19
  "andgrt/layoutlmv2-base-uncased_finetuned_docvqa"
20
  )
21
+
22
+ git_model_base = AutoModelForDocumentQuestionAnswering.from_pretrained(
23
  "andgrt/layoutlmv2-base-uncased_finetuned_docvqa"
24
  )
25