Spaces:

atnanahidiw
/

brand-new-space

Runtime error

atnanahidiw commited on Mar 24, 2024

Commit

fb7a6d8

verified ·

1 Parent(s): f7c1ca9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ processor = DonutProcessor.from_pretrained("nielsr/donut-demo")
 model = VisionEncoderDecoderModel.from_pretrained("nielsr/donut-demo")
-def donut(input_img):
     # prepare encoder inputs
     pixel_values = processor(sample["image"].convert("RGB"), return_tensors="pt").pixel_values
     pixel_values = pixel_values.to(device)

 model = VisionEncoderDecoderModel.from_pretrained("nielsr/donut-demo")
+def donut(sample):
     # prepare encoder inputs
     pixel_values = processor(sample["image"].convert("RGB"), return_tensors="pt").pixel_values
     pixel_values = pixel_values.to(device)