gagan3012 commited on
Commit
2a35e7d
·
1 Parent(s): 755cdec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -59,6 +59,7 @@ def predict_english(img, model_name="naver-clova-ix/donut-base-finetuned-cord-v2
59
  return sequence
60
 
61
  def predict_nougat(img, model_name="facebook/nougat-small"):
 
62
  processor = NougatProcessor.from_pretrained(model_name)
63
  model = VisionEncoderDecoderModel.from_pretrained(model_name)
64
  image = img.convert("RGB")
 
59
  return sequence
60
 
61
  def predict_nougat(img, model_name="facebook/nougat-small"):
62
+ device="cpu"
63
  processor = NougatProcessor.from_pretrained(model_name)
64
  model = VisionEncoderDecoderModel.from_pretrained(model_name)
65
  image = img.convert("RGB")