Update app.py

app.py

@@ -17,6 +17,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
 def process_document(image):
+    img2.update(visible=True)
     #can't save uploaded file locally, but needs to be converted from nparray to PIL
     im1 = Image.fromarray(image)
     
@@ -57,7 +58,8 @@ def process_document(image):
     sequence = processor.batch_decode(outputs.sequences)[0]
     sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(processor.tokenizer.pad_token, "")
     sequence = re.sub(r"<.*?>", "", sequence, count=1).strip()  # remove first task start token
-    
+
+    img2.update(visible=False)
     return processor.token2json(sequence), image
 
 title = '<table align="center" border="0" cellpadding="1" cellspacing="1" style="width:100pc"><tbody><tr><td style="text-align:center"><img alt="" src="https://huggingface.co/spaces/to-be/invoice_document_headers_extraction_with_donut/resolve/main/circling_small.gif" style="float:right; height:50px; width:50px" /></td><td style="text-align:center"><h1>&nbsp; &nbsp;Welcome</h1></td><td style="text-align:center"><img alt="" src="https://huggingface.co/spaces/to-be/invoice_document_headers_extraction_with_donut/resolve/main/circling2_small.gif" style="float:left; height:50px; width:50px" /></td></tr></tbody></table>'
@@ -88,7 +90,7 @@ with gr.Blocks(css=css) as demo:
              gr.Examples([["example.jpg"], ["example_2.jpg"], ["example_3.jpg"]], inputs=[inp],label='Or use one of these examples:')
     with gr.Row().style(equal_height=True,height=200,rounded=False):       
         with gr.Column(scale=1):
-            img2 = gr.Image("drinking.gif",label=' ').style(rounded=True)
+            img2 = gr.Image("drinking.gif",label=' ',visible=False).style(rounded=True)
         with gr.Column(scale=2):
             btn = gr.Button("↓   Extract   ↓")
         with gr.Column(scale=1):