update code 2

app.py

@@ -44,9 +44,11 @@ def get_dark_mode():
     }
     """
 
-def clear_session():
+def clear_session(image):
     global clear_flag 
     clear_flag = True
+    if image is not None:
+        return gr.update(value=None), gr.update(value=None)
     return gr.update(value=None), gr.update(value=None)
 
 
@@ -317,7 +319,7 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
         outputs=outputs,
         api_name=False,
     )
-    clear_button.click(fn=clear_session, inputs=[], outputs=[image, result])
+    clear_button.click(fn=clear_session, inputs=[image], outputs=[image, result])
     val_r.change(assign_gpu, inputs=inputs, outputs=outputs, queue=False, api_name=False)
     run_button.click(fn=assign_gpu, inputs=inputs, outputs=outputs, api_name=False)
     image.change(assign_gpu, inputs=inputs, outputs=outputs, queue=False, api_name=False)