File size: 971 Bytes
8165e64 82fdd29 8165e64 0116b88 98dbc07 acce120 8165e64 41944f2 8165e64 0116b88 98dbc07 8165e64 230dc89 4895857 230dc89 8165e64 |
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 26 27 28 29 30 |
import gradio as gr
from gradio_client import Client
import spaces
def text_to_image(prompt):
from gradio.context import LocalContext
request = LocalContext.request.get(None)
print(f"Request available in text_to_image: {request is not None} {request}")
print("x-ip", request.headers["x-ip-token"])
client = Client("hysts/SDXL")
img = client.predict(prompt, api_name="/predict")
return img
@spaces.GPU
def generate(prompt):
from gradio.context import LocalContext
request = LocalContext.request.get(None)
print(f"Request available in generate: {request is not None} {request}")
print("x-ip", request.headers["x-ip-token"])
return text_to_image(prompt)
with gr.Blocks() as demo:
image = gr.Image()
prompt = gr.Textbox(max_lines=1)
button = gr.Button("Chain")
prompt.submit(generate, [prompt], [image])
button.click(lambda x:x+x, button, button).then(generate, [prompt], [image])
demo.launch()
|