Update app.py

app.py

@@ -6,16 +6,15 @@ def set_client_for_session(request: gr.Request):
     x_zero_gpu_id = request.headers["x-zerogpu-uuid"]
     return Client("hysts/SDXL", headers={"x-zerogpu-token": x_gpu_token, "x-zero-gpu-uuid": x_zero_gpu_id})
 
-def text_to_image(prompt, client):
+def text_to_image(prompt, request: gr.Request):
+    client = set_client_for_session(request)
     img = client.predict(prompt, "", "", "", api_name="/run")
     return img
 
 with gr.Blocks() as demo:
-    client = gr.State()
     image = gr.Image()
     prompt = gr.Textbox(max_lines=1)
     prompt.submit(text_to_image, [prompt, client], [image])
-    demo.load(set_client_for_session, inputs=None, outputs=client)
 
 demo.launch()