File size: 2,414 Bytes
6634db1 83300eb 42ec9a0 bc48fae 500a5b0 4cb7e61 8f4468f 9358b29 c84a6e6 c4f9926 1bef23c 6614cbc 00084b4 42ec9a0 83300eb 9358b29 bc48fae 00084b4 2184e90 731f50d b1ea448 64ca3f9 b1ea448 64ca3f9 b1ea448 88bc3d1 b21c75d 75a1978 42ec9a0 b21c75d 83300eb 6614cbc 83300eb 1bef23c c2de16e 0c30387 42ec9a0 fbcabeb 2e84177 42ec9a0 e955094 fbcabeb 6634db1 |
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 60 61 62 63 64 65 66 67 68 69 70 |
import gradio as gr
import diff
import os
from PIL import Image
import uuid
sky = "https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/sky.png"
platform = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/platform.png'
star = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/star.png'
bomb = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/bomb.png'
dude = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/dude.png'
models=[
gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base",live=True)
]
def game_fn(sky=sky,platform=platform,star=star,bomb=bomb,dude=dude):
html_mod=f"""
<div id="demo" style="height:600px">
<iframe
id="myIframe"
src="https://omnibus-game-test-static.static.hf.space/index.html?sky={sky}&platform={platform}&star={star}&bomb={bomb}&dude={dude}"
frameborder="0"
width="100%"
height="100%"
></iframe>
</div>"""
return html_mod
def update_game():
return game_fn(sky=sky)
def dif_fn(inp):
#dif = gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base", live=True),
dif = models[0]
uid=uuid.uuid4()
#output=diff.send_it(inp,5,1)
output=dif(inp)
print(output)
outp=Image.open(output)
width, height = outp.size
rat = width/height
if width > height:
outp = outp.resize((600*rat,600))
elif width < height:
outp = outp.resize((800,800*rat))
else:
outp = outp.resize((800,536))
outp.save(f"{uid}_sky.png")
out = os.path.abspath(f"{uid}_sky.png")
out = os.path.abspath(outp)
out_url = f'https://omnibus-game-test.hf.space/file={out}'
return outp,out_url
with gr.Blocks() as app:
with gr.Row():
with gr.Column():
prompt=gr.Textbox()
with gr.Row():
btn=gr.Button("Make Image")
update_game=gr.Button("Use Image")
out_im=gr.Image(type='filepath')
out_url=gr.Textbox(visible=False)
start_prompt=gr.Textbox(value="A beautiful landscape",visible=False)
html_game = gr.HTML()
update_game.click(game_fn,[out_url],html_game)
btn.click(dif_fn,prompt,[out_im,out_url])
app.load(dif_fn,start_prompt,[out_im,out_url]).then(game_fn,[out_url],html_game)
app.launch() |