Spaces:
Sleeping
Sleeping
File size: 1,761 Bytes
d8262b8 34e1d28 d8262b8 2d81e11 67ef4bf d8262b8 2d81e11 fce7b46 67ef4bf d8262b8 67ef4bf 69dac1f 34a801c |
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 |
import gradio as gr
from controlnet_aux import OpenposeDetector
from PIL import Image
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
import torch
from controlnet_aux import OpenposeDetector
from diffusers.utils import load_image
#Models
openpose = OpenposeDetector.from_pretrained('lllyasviel/ControlNet')
controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch.float16)
pipe = StableDiffusionControlNetPipeline.from_pretrained("helkoo/jelaba_2HR", controlnet=controlnet, safety_checker=None, torch_dtype=torch.float16)
#optimizations
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
pipe = pipe.to("cpu")
import numpy as np
import requests
def generate2(prompt,taille):
if taille == "S":
image = Image.open(requests.get('https://mode-et-caftan.com/757-large_default/jellaba-salsa-marocaine-femme.jpg', stream=True).raw)
if taille == "XL":
image = Image.open(requests.get('https://i.pinimg.com/236x/03/f1/36/03f136b83bb37c9f17c3764f1b36f9fa--big-is-beautiful-curvy-fashion.jpg', stream=True).raw)
if taille == "L":
image = Image.open(requests.get('https://mode-et-caftan.com/757-large_default/jellaba-salsa-marocaine-femme.jpg', stream=True).raw)
# convert image to numpy array
image = np.array(image)
image = openpose(image)
#image = image
image = pipe(prompt, image, num_inference_steps=20).images[0]
return image
gr.Interface(fn=generate2, inputs=["text",
gr.Dropdown(
["S", "L", "XL"], label="taille", info="choisie la taille"
),
], outputs="image").launch(share=False, debug=True)
|