Spaces:

Donato8408
/

impira-layoutlm-document-qa

Runtime error

Donato8408 commited on Mar 27, 2023

Commit

3eec87b

•

1 Parent(s): c1c6489

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ from transformers import AutoTokenizer, AutoModelForQuestionAnswering
 import torch
 # Load model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("microsoft/layoutlm-base-uncased")
-model = AutoModelForQuestionAnswering.from_pretrained("microsoft/layoutlm-base-uncased")
 # Define function to predict answer
 def predict_answer(context, question):

 import torch
 # Load model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained("impira/layoutlm-document-qa")
+model = AutoModelForQuestionAnswering.from_pretrained("impira/layoutlm-document-qa")
 # Define function to predict answer
 def predict_answer(context, question):