Update app.py

app.py

@@ -16,6 +16,8 @@ model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
+state = "start_or_clear"
+
 def process_document(image):
     img2.update(visible=True)
     gr.update()
@@ -101,7 +103,19 @@ with gr.Blocks(css=css) as demo:
             imgout = gr.Image(label='Uploaded document:',elem_id="inp")
         with gr.Column(scale=1):
             jsonout = gr.JSON(label='Extracted information:')
-            
+    inp.change(fn=update_status,inputs=inp,outputs=[img2,img3])        
     btn.click(fn=process_document, inputs=inp, outputs=[jsonout,imgout])
 
-demo.launch()
+demo.launch()
+def update_status(inp):
+    if state == "start_or_clear": 
+        state = 'processing'   #current state becomes
+        return (gr.update(value="snowangel.gif",visible=True),gr.update(value="snowangel.gif",visible=True))
+    elif state == "processing":
+        state = 'finished_processing' #current state becomes
+        return (gr.update(value="",visible=False),gr.update(value="",visible=False))
+    elif state == "finished_processing":
+        state = 'processing'   #current state becomes
+        return (gr.update(value="snowangel.gif",visible=True),gr.update(value="snowangel.gif",visible=True))
+
+