Update app.py

app.py

@@ -16,9 +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 update_status(inp):
+def update_status(state):
     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))
@@ -91,6 +90,7 @@ css = "#inp {height: auto !important; width: 100% !important;}"
 
 
 with gr.Blocks(css=css) as demo:
+    state = gr.State(value='start_or_clear')
     
     gr.HTML(title)
     gr.HTML(paragraph1)
@@ -114,7 +114,7 @@ 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:')
-    imgout.change(fn=update_status,inputs=inp,outputs=[img2,img3])        
+    imgout.change(fn=update_status,inputs=state,outputs=[img2,img3])        
     btn.click(fn=process_document, inputs=inp, outputs=[jsonout,imgout])
 
 demo.launch()