add model checkpoint revision as an option

app.py

@@ -86,9 +86,10 @@ def translate_point_coords_from_out_to_in(point=None, input_image_size=None, out
         f"translated point={point}, resized_image_size: {resized_width, resized_height}")
 
 
-def process_refexp(image: Image, prompt: str):
+def process_refexp(image: Image, prompt: str, revision: str = 'main'):
 
     print(f"(image, prompt): {image}, {prompt}")
+    print(f"model checkpoint revision: {revision}")
 
     # trim prompt to 80 characters and normalize to lowercase
     prompt = prompt[:80].lower()
@@ -184,7 +185,7 @@ def process_refexp(image: Image, prompt: str):
 
 
 title = "Demo: Donut 🍩 for UI RefExp - Center Point (by GuardianUI)"
-description = "Gradio Demo for Donut RefExp task, an instance of `VisionEncoderDecoderModel` fine-tuned on [UIBert RefExp](https://huggingface.co/datasets/ivelin/ui_refexp_saved) Dataset (UI Referring Expression). To use it, simply upload your image and type a prompt and click 'submit', or click one of the examples to load them. See the model training <a href='https://colab.research.google.com/github/ivelin/donut_ui_refexp/blob/main/Fine_tune_Donut_on_UI_RefExp.ipynb' target='_parent'>Colab Notebook</a> for this space. Read more at the links below."
+description = "Gradio Demo for Donut RefExp task, an instance of `VisionEncoderDecoderModel` fine-tuned on [UIBert RefExp](https://huggingface.co/datasets/ivelin/ui_refexp_saved) Dataset (UI Referring Expression). To use it, simply upload your image and type a prompt and click 'submit', or click one of the examples to load them. Optionally enter value for model git revision; latest checkpoint will be used by default."
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.15664' target='_blank'>Donut: OCR-free Document Understanding Transformer</a> | <a href='https://github.com/clovaai/donut' target='_blank'>Github Repo</a></p>"
 examples = [["example_1.jpg", "select the setting icon from top right corner"],
             ["example_1.jpg", "click on down arrow beside the entertainment"],
@@ -211,7 +212,7 @@ examples = [["example_1.jpg", "select the setting icon from top right corner"],
             ]
 
 demo = gr.Interface(fn=process_refexp,
-                    inputs=[gr.Image(type="pil"), "text"],
+                    inputs=[gr.Image(type="pil"), "text", "text"],
                     outputs=[gr.Image(type="pil"), "json"],
                     title=title,
                     description=description,