Update app.py

app.py

@@ -1,22 +1,22 @@
 import gradio as gr
 from gradio_client import Client
 
-client = Client("hysts/SDXL")
-
-def text_to_image(prompt):
-    print(">>>>>>>>MMMMMMMM")
-    from gradio.context import LocalContext
-    print("localcontext", LocalContext.request.get())
-    img = client.predict(prompt, "", "", "", api_name="/run")
-    print("<<<<<<<<NNNNNNNN")
+def text_to_image(client, prompt):
+    img = client.predict(prompt, api_name="/predict")
     return img
 
+
+def set_client_for_session(request: gr.Request):
+    x_ip_token = request.headers['x-ip-token']
+
+    # The "gradio/text-to-image" space is a ZeroGPU space
+    return Client("gradio/text-to-image", headers={"X-IP-Token": x_ip_token})
+
 with gr.Blocks() as demo:
+    client = gr.State()
     image = gr.Image()
     prompt = gr.Textbox(max_lines=1)
-    prompt.submit(text_to_image, [prompt], [image])
 
-demo.launch()
+    prompt.submit(text_to_image, [client, prompt], [image])
 
-# demo = gr.load("spaces/hysts/SDXL")
-# demo.launch()
+    demo.load(set_client_for_session, None, client)