Update app.py

app.py

@@ -1,3 +1,23 @@
 import gradio as gr
+from gradio_client import Client
 
-gr.load("hysts/SDXL", src="spaces").launch()
+def text_to_image(client, prompt):
+    img = client.predict(prompt, "", "", "", api_name="/run")
+    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("hysts/SDXL", 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, [client, prompt], [image])
+
+    demo.load(set_client_for_session, None, client)
+demo.launch()