Update app.py

app.py

@@ -8,6 +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")
@@ -49,7 +50,7 @@ def predict(image):
   return page_sequence
 
 
-
+@spaces.GPU
 def inference(pdf_file, pdf_link):
   if pdf_file is None:
     if pdf_link == '':