Update app.py
Browse files
app.py
CHANGED
|
@@ -24,50 +24,10 @@ def run_hidiffusion(prompt, negative_prompt="", progress=gr.Progress(track_tqdm=
|
|
| 24 |
images.append(result.images[0])
|
| 25 |
return images
|
| 26 |
|
| 27 |
-
@spaces.GPU
|
| 28 |
-
def run_hidiffusion_15(prompt, negative_prompt="", progress=gr.Progress(track_tqdm=True)):
|
| 29 |
-
apply_hidiffusion(pipe_15)
|
| 30 |
-
images = []
|
| 31 |
-
for _ in range(2): # 4개의 이미지 생성
|
| 32 |
-
result = pipe_15(prompt, guidance_scale=7.5, height=1024, width=1024, eta=1.0, negative_prompt=negative_prompt, num_inference_steps=25)
|
| 33 |
-
images.append(result.images[0])
|
| 34 |
-
return images
|
| 35 |
-
|
| 36 |
|
| 37 |
with gr.Blocks() as demo:
|
| 38 |
gr.Markdown("# AI Human Model Generator")
|
| 39 |
-
|
| 40 |
-
with gr.Tab("1024x1024"):
|
| 41 |
-
with gr.Row():
|
| 42 |
-
prompt_15 = gr.Textbox(label="Prompt")
|
| 43 |
-
negative_prompt_15 = gr.Textbox(
|
| 44 |
-
label="Negative Prompt",
|
| 45 |
-
value="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, (NSFW:1.25)",
|
| 46 |
-
visible=False
|
| 47 |
-
)
|
| 48 |
-
btn_15 = gr.Button("Run")
|
| 49 |
-
output_gallery_15 = gr.Gallery(label="Results")
|
| 50 |
-
btn_15.click(fn=run_hidiffusion_15, inputs=[prompt_15, negative_prompt_15], outputs=[output_gallery_15])
|
| 51 |
-
|
| 52 |
-
gr.Examples(examples=[
|
| 53 |
-
"a beautiful model woman, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 54 |
-
"a beautiful model woman, 20 year aged, Caucasian ethnicity, nurse uniform attire, full body visible, model pose, direct frontal gaze, white color background, realritics photo,High Quaility, UHD",
|
| 55 |
-
"a beautiful model woman, Caucasian ethnicity, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 56 |
-
"a beautiful model woman, Black ethnicity, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 57 |
-
"a beautiful model woman, Caucasian ethnicity,Business attire, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 58 |
-
"a beautiful model woman, Caucasian ethnicity,casual attire, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 59 |
-
"a beautiful model woman, golf wear attire, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 60 |
-
"a handsome model man, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 61 |
-
"a handsome model man, Caucasian ethnicity,uniform attire, full body visible, model pose, suite wear, direct frontal gaze, blue color background, realritics photo, High Quaility, UHD",
|
| 62 |
-
"a handsome model man, Business attire, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 63 |
-
"a handsome model man, Caucasian ethnicity, full body visible, model pose, direct frontal gaze, white color background, realritics photo, 16k",
|
| 64 |
-
"a handsome model man, Caucasian ethnicity,Business attire, Black ethnicity, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 65 |
-
"a handsome model man, casual attire, pilot uniform attire, full body visible, model pose, direct frontal gaze, white color background, realritics photo, High Quaility, UHD",
|
| 66 |
-
"a handsome model man, Caucasian ethnicity,golf wear attire, full body visible, model pose, direct frontal gaze, gray color background, realritics photo, High Quaility, UHD"
|
| 67 |
-
],
|
| 68 |
-
inputs=[prompt_15],
|
| 69 |
-
outputs=[output_gallery_15]) # 변수 이름을 올바르게 수정
|
| 70 |
-
|
| 71 |
with gr.Tab("2048x2048"):
|
| 72 |
with gr.Row():
|
| 73 |
prompt = gr.Textbox(label="Prompt")
|
|
|
|
| 24 |
images.append(result.images[0])
|
| 25 |
return images
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
with gr.Blocks() as demo:
|
| 29 |
gr.Markdown("# AI Human Model Generator")
|
| 30 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
with gr.Tab("2048x2048"):
|
| 32 |
with gr.Row():
|
| 33 |
prompt = gr.Textbox(label="Prompt")
|