Change model_state to get function instead of calling it

app.py

@@ -144,7 +144,7 @@ def diffusion_chat(question, eot_weight, max_it, sharpness, model):
     yield f"<b>Final Output (after {i+1} iterations):</b><br>" + final_output
 
 # --- Gradio Interface ---
-model_state = gr.State(load_model())
+model_state = gr.State(value=load_model())  # this just stores the object
 
 demo = gr.Interface(
     fn=diffusion_chat,