Update app.py

app.py

@@ -280,6 +280,9 @@ css = """
 
 # Build Gradio UI
 with gr.Blocks(css=css) as demo:
+    session_state = gr.State()
+    demo.load(start_session, outputs=[session_state])
+
     gr.HTML(
         """
         <div style="text-align: center;">
@@ -319,14 +322,12 @@ with gr.Blocks(css=css) as demo:
         
                 examples = gr.Examples(
                     examples=["example_images/jim_carrey.png", "example_images/margaret_qualley.png"],
-                    inputs=[image_in],
+                    inputs=[image_in, session_state],
                     outputs=[],  # outputs are specified via the callback
                     fn=generate_results_and_mesh,
                     cache_examples=True  # cache results of examples for faster loading
                 )
 
-        session_state = gr.State()
-        demo.load(start_session, outputs=[session_state])
 
         run_btn.click(
             fn=generate_results_and_mesh,