Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|