Update app.py

app.py

@@ -8,7 +8,7 @@ import torch
 import gradio as gr
 import uuid
 import os
-
+import spaces
 processor = NougatProcessor.from_pretrained("facebook/nougat-small")
 model = VisionEncoderDecoderModel.from_pretrained("facebook/nougat-small")
 
@@ -30,7 +30,7 @@ def get_pdf(pdf_link):
   return unique_filename
 
 
-
+@spaces.GPU
 def predict(image):
   # prepare PDF image for the model
   image = Image.open(image)