i-dont-hug-face
/

adamcodd-donut-extract

vision-encoder-decoder

Model card Files Files and versions Community

i-dont-hug-face commited on Jul 11

Commit

26c91ca

•

1 Parent(s): af679a9

Update inference.py

Files changed (1) hide show

inference.py +1 -0

inference.py CHANGED Viewed

@@ -6,6 +6,7 @@ import io
 import json
 def model_fn(model_dir, context=None):
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     processor = DonutProcessor.from_pretrained(model_dir)
     model = VisionEncoderDecoderModel.from_pretrained(model_dir)

 import json
 def model_fn(model_dir, context=None):
+    print("this is model_dir: ", model_dir)
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     processor = DonutProcessor.from_pretrained(model_dir)
     model = VisionEncoderDecoderModel.from_pretrained(model_dir)