minDalle_GFPGAN / app.py
Logan Zoellner
try this
1b121c9
raw
history blame
1.24 kB
from asyncio import constants
import gradio as gr
import requests
import os
from base64 import b64decode
from PIL import Image
import io
def generate_image(text):
#dalle = gr.Interface.load("spaces/kuprel/min-dalle")
dalle = gr.Interface.load("spaces/Axolotlily/DalleMini")
print("calling interface",text)
img=dalle(text)
#img=dalle.fns[0].fn(seed,psi)
#header, encoded = img.split(",", 1)
#data = b64decode(encoded)
#image = Image.open(io.BytesIO(data))
return img
#gfpgan=gr.Interface.load("spaces/akhaliq/GFPGAN")
#img2=dalle(image)
#return img2
demo = gr.Blocks()
with demo:
gr.Markdown("<h1><center>StyleGan-Human + PIFu </center></h1>")
gr.Markdown(
"create an image with min-dalle then fix faces with grpgan"
)
with gr.Row():
b0 = gr.Button("generate image")
with gr.Row():
text=gr.Text(default="three pigs in a trenchcoat", label='Seed')
#outputImage = gr.Image(label="portrait",type="filepath", shape=(256,256))
output_image = gr.outputs.Image(type="filepath", label='Output')
b0.click(generate_image,inputs=[text],outputs=[output_image])
demo.launch(enable_queue=True, debug=True)