Update app.py

app.py

@@ -1,5 +1,6 @@
 import re
 import gradio as gr
+import spaces
 
 import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
@@ -10,6 +11,7 @@ model = VisionEncoderDecoderModel.from_pretrained("naver-clova-ix/donut-base-fin
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
+@spaces.GPU()
 def process_document(image):
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values