Add GPU support for text extraction

app.py

@@ -4,6 +4,7 @@ import torch
 from PIL import Image
 from pathlib import Path
 from pdf2image import convert_from_path
+import spaces
 
 # Load the model and processor
 processor = NougatProcessor.from_pretrained("MohamedRashad/arabic-small-nougat")
@@ -14,6 +15,7 @@ model.to(device)
 print(f"Using {device} device")
 context_length = 2048
 
+@spaces.GPU
 def extract_text_from_image(image):
     """
     Extract text from PIL image