|
|
|
import numpy as np |
|
import gradio as gr |
|
|
|
def inmm(init_image, prompt): |
|
(w,h) = init_image.size |
|
if w>h : |
|
init_image = init_image.crop(((w - h) // 2,0,(w-h)//2 + h,h)) |
|
init_image = init_image.resize(512,512) |
|
|
|
with autocast("cuda"): |
|
image = pipe([prompt], num_inference_steps=50, guidance_scale=7, init_image=init_image)["sample"] |
|
return image[0] |
|
|
|
demo = gr.Interface(inmm, inputs=[gr.Image(shape=(512, 512),type="pil"),gr.Textbox(lines=2, placeholder="どんな絵が欲しいか",value ="a heartwarming and calming landscape drawing in <m-mi2i> style")], outputs="image", |
|
examples=[["a_img.png", "A heartwarming Canadian wheat field scene in <m-mi2i> style, some houses, silos, and a lake in the distance"], |
|
["c_img.png","A heartwarming Landscape on the lake, scenery mirrored on the lake, in <m-mi2i> style"]]) |
|
demo.launch() |