freddyaboulton's picture
Update app.py
cec08f8 verified
raw
history blame
752 Bytes
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()