Update device selection for model

app.py

@@ -1,11 +1,12 @@
 from transformers import NougatProcessor, VisionEncoderDecoderModel
 import gradio as gr
+import torch
 from pdf2image import convert_from_path
 
 # Load the model and processor
 processor = NougatProcessor.from_pretrained("MohamedRashad/arabic-small-nougat")
 model = VisionEncoderDecoderModel.from_pretrained("MohamedRashad/arabic-small-nougat")
-device = "cpu"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 context_length = 2048