Upadated with random example

app.py

@@ -1,14 +1,14 @@
 import re
 import gradio as gr
 
-# import torch
+import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
 
 processor = DonutProcessor.from_pretrained("Travad98/donut-finetuned-sogc-trademarks-1883-2001")
 model = VisionEncoderDecoderModel.from_pretrained("Travad98/donut-finetuned-sogc-trademarks-1883-2001")
 
-# device = "cuda" if torch.cuda.is_available() else "cpu"
-# model.to(device)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model.to(device)
 
 def process_document(image):
     # prepare encoder inputs
@@ -50,7 +50,7 @@ demo = gr.Interface(
     description=description,
     article=article,
     enable_queue=True,
-    examples=[["example.png"], ["example_2.png"], ["example_3.png"]],
+    examples=[["sha-001_1883_1__21_d-0-6.jpg"]],
     cache_examples=False)
 
 demo.launch()