Update app.py

app.py

@@ -3,9 +3,14 @@ import gradio as gr
 
 import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
+radio="debu-das/donut_receipt_v1.20"
 
-processor = DonutProcessor.from_pretrained("debu-das/donut_receipt_v1.20")
-model = VisionEncoderDecoderModel.from_pretrained("debu-das/donut_receipt_v1.20")
+radio = gr.Radio(
+        ["debu-das/donut_receipt_v1.20", "debu-das/donut_receipt_v1.21", "debu-das/donut_receipt_v1.22","debu-das/donut_receipt_v1.23","debu-das/donut_receipt_v1.24",], label="select the model version "
+    )
+
+processor = DonutProcessor.from_pretrained(radio)
+model = VisionEncoderDecoderModel.from_pretrained(radio)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)