Spaces:
Sleeping
Sleeping
# ============================== | |
# Basic Interface | |
# ============================== | |
#import gradio as gr | |
#demo = gr.load("mswhite/miamivice", src="models") | |
#demo.launch() | |
# ============================== | |
# Detailed Interface | |
# ============================== | |
import os | |
import gradio as gr | |
import torch | |
from diffusers import DiffusionPipeline, StableDiffusionPipeline, StableDiffusionImg2ImgPipeline | |
# Nvidia GPU config | |
#from diffusers import DiffusionPipeline | |
#pipe = DiffusionPipeline.from_pretrained("mswhite/nnn-model") | |
pipe = StableDiffusionPipeline.from_pretrained("mswhite/bricksburg_model", torch_dtype=torch.float16) #, safety_checker=None) | |
pipe = pipe.to("cuda") | |
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images)) | |
# CPU config | |
#pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice") | |
# Optional Img2Img pipeline | |
#pipe_img = StableDiffusionImg2ImgPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16, safety_checker=None) | |
#pipe_img = pipe_img.to("cuda") | |
# Sample Prompt - "bright clean clear illustration side profile of a woman wearing pink lipstick p_nagel" | |
def nagelize(text): | |
prompt = "mini_mw_lgo" + text | |
return pipe(prompt).images[0] | |
#def nagelize_me(source_image): | |
# prompt = "p_nagel" | |
# return pipe_img(prompt, image=source_image, strength=0.6, num_inference_steps=500, guidance_scale=40.0).images[0] | |
# Web Version of Blocks() | |
#with gr.Blocks() as demo: | |
#girl_image = gr.Image(label = None, value = os.path.join(os.path.dirname(__file__), "headshot.jpg"), visible=False) | |
# with gr.Row(): | |
# logo_image = gr.Image(label = None, value = os.path.join(os.path.dirname(__file__), "miami.jpg"), interactive=True).style(height=720) | |
# #gr.Markdown("<img src=https://huggingface.co./spaces/mswhite/miamivice/resolve/main/miami.jpg width=720px>") | |
# output = gr.Image(label="Output").style(height=720) | |
# with gr.Row(): | |
# art_to_draw = gr.Textbox(label="Prompt to Draw: e.g. woman with red hair") | |
# flashb_btn = gr.Button("Flashback to the 80s") | |
# #back80_btn = gr.Button("Send me back to the 80s") | |
# flashb_btn.click(fn=nagelize, inputs=[art_to_draw], outputs=[output]) | |
# #back80_btn.click(fn=nagelize, inputs=[art_to_draw], outputs=[output]) | |
# Mobile Version of Blocks | |
with gr.Blocks() as demo: | |
gr.Markdown("<img src=https://huggingface.co./spaces/mswhite/Bricksburg/resolve/main/Emmet.jpg width=720px>") | |
output = gr.Image(label="Output").style(width=720) | |
art_to_draw = gr.Textbox(label="Prompt to Draw: e.g. ") | |
flashb_btn = gr.Button("Everything is Awesome in Bricksburg") | |
flashb_btn.click(fn=nagelize, inputs=[art_to_draw], outputs=[output]) | |
demo.launch() | |