Logan Zoellner
fix indent
326480a
raw
history blame
1.2 kB
from asyncio import constants
import gradio as gr
import requests
import os
def generate_image(seed,psi):
iface = gr.Interface.load("spaces/hysts/StyleGAN-Human")
img=iface(seed,psi)
return img
def generate_model(img):
iface = gr.Interface.load("spaces/radames/PIFu-Clothed-Human-Digitization/tree/main")
model,file=iface(img)
return model,file
demo = gr.Blocks()
with demo:
gr.Markdown("<h1><center>StyleGan-Human + PIFu </center></h1>")
gr.Markdown(
"create a person and then generate a model from that person's image"
)
with gr.Row():
b0 = gr.Button("generate image")
b1 = gr.Button("generate model")
with gr.Row():
seed=gr.Number(default=0, label='Seed'),
psi=gr.Slider(0, 2, step=0.05, default=0.7, label='Truncation psi')
img = gr.Image(label="portrait",type="filepath", shape=(256,256))
model = gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0], label="3D Model"),
file= gr.File(label="Download 3D Model")
b0.click(generate_image,inputs=[seed,psi],outputs=img)
b1.click(generate_model, inputs=img, outputs=[model,file])
demo.launch(enable_queue=True, debug=True)