File size: 752 Bytes
2dbca95
06c9b54
2dbca95
21a4d99
6275c42
375a394
e3de9d5
96c5330
cec08f8
00a4341
06c9b54
e3de9d5
06c9b54
6275c42
06c9b54
375a394
06c9b54
 
08168f0
375a394
6275c42
06c9b54
6275c42
06c9b54
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import gradio as gr
from gradio_client import Client

# client = Client("hysts/SDXL")

def text_to_image(prompt, client):
    print(">>>>>>>>MMMMMMMM")
    from gradio.context import LocalContext
    print("localcontext.request", LocalContext.request)
    print("localcontext", LocalContext.request.get().headers.get("x-ip-token"))
    img = client.predict(prompt, "", "", "", api_name="/run")
    print("<<<<<<<<NNNNNNNN")
    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(lambda: Client("hysts/SDXL"), inputs=None, outputs=client)

demo.launch()

# demo = gr.load("spaces/hysts/SDXL")
# demo.launch()