Spaces:
Runtime error
Runtime error
File size: 1,795 Bytes
1c0019e 6f62fb6 c3ccb1a 1c0019e 6f62fb6 1c0019e 6f62fb6 1c0019e c3ccb1a 1c0019e 6f62fb6 1c0019e bea8d25 1c0019e 7252988 7bfda4a 5052c63 7252988 1c0019e cd2d865 de7878e cd2d865 de7878e 1c0019e |
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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import gradio as gr
import random
import requests
import io
from PIL import Image
import os
"""
API_URL = "https://api-inference.huggingface.co/models/LucyintheSky/lucy-dream-style"
headers = {"Authorization": "Bearer " + os.environ.get('TOKEN')}
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
"""
def generate(prompt):
"""
image_bytes = query({
"inputs": "lucy dream style " + prompt,
"parameters" : { "negative_prompt": "ugly, deformed, bad quality",
"seed": random.randint(0,9999999)}
})
image = Image.open(io.BytesIO(image_bytes))
return image
"""
return None
theme = gr.themes.Base(
primary_hue="gray",
secondary_hue="gray",
neutral_hue="gray",
font=['Helvetica', 'ui-sans-serif', 'system-ui', 'sans-serif'],
).set(
button_large_text_weight='400',
input_background_fill='#ffffff',
#input_border_width='*block_border_width',
button_primary_background_fill='#ffd8db',
#button_border_width='*block_border_width',
)
with gr.Blocks(theme=theme) as demo:
gr.Markdown("""
<center> <img src="https://cdn-uploads.huggingface.co/production/uploads/650a1281b48ff3906647edd0/rbzPKW_p5XTG9leEO4uFt.png" > </center>
<br><br><br>
""")
img = gr.Image(show_label=False, type='pil')
textbox = gr.Textbox(show_label=False, placeholder='type your prompt in here')
button = gr.Button("generate", variant="primary")
gr.Markdown("""
""")
button.click(fn=generate, inputs=textbox, outputs=img)
textbox.submit(fn=generate, inputs=textbox, outputs=img)
demo.launch() |