File size: 4,424 Bytes
61bbe53
f14200d
61bbe53
0c7479d
f14200d
0c7479d
 
 
f14200d
61bbe53
babfe4f
 
 
 
 
 
 
2474e74
f14200d
 
 
 
 
61bbe53
 
 
f14200d
0c7479d
 
 
f14200d
2474e74
f14200d
 
2474e74
f14200d
 
 
 
 
 
2474e74
f14200d
bc1676d
 
f14200d
2474e74
02dd2e1
 
 
 
 
 
 
 
0c7479d
 
 
 
 
 
 
 
 
 
11a775c
 
f14200d
 
61bbe53
2474e74
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import gradio as gr
import os

from editany import create_demo as create_demo_edit_anything
from sam2image import create_demo as create_demo_generate_anything
from editany_beauty import create_demo as create_demo_beauty
from editany_handsome import create_demo as create_demo_handsome
from editany_lora import EditAnythingLoraModel, init_sam_model, init_blip_processor, init_blip_model
from huggingface_hub import hf_hub_download, snapshot_download

DESCRIPTION = f'''# [Edit Anything](https://github.com/sail-sg/EditAnything)
**Edit anything and keep the layout by segmenting anything in the image.**
'''
SHARED_UI_WARNING = f'''### [NOTE]  Inference may be slow in this shared UI.
You can duplicate and use it with a paid private GPU.
<a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/jyseo/3DFuse?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-xl-dark.svg" alt="Duplicate Space"></a>
'''
sam_generator, mask_predictor = init_sam_model()
blip_processor = init_blip_processor()
blip_model = init_blip_model()

sd_models_path = snapshot_download("shgao/sdmodels")

with gr.Blocks() as demo:
    gr.Markdown(DESCRIPTION)
    with gr.Tabs():
        with gr.TabItem('🖌Edit Anything'):
            model = EditAnythingLoraModel(base_model_path="runwayml/stable-diffusion-v1-5",
                                          controlmodel_name='LAION Pretrained(v0-4)-SD15',
                                          lora_model_path=None, use_blip=True, extra_inpaint=True,
                                          sam_generator=sam_generator,
                                          mask_predictor=mask_predictor,
                                          blip_processor=blip_processor,
                                          blip_model=blip_model)
            create_demo_edit_anything(model.process, model.process_image_click)
        with gr.TabItem(' 👩‍🦰Beauty Edit/Generation'):
            lora_model_path = hf_hub_download(
                "mlida/Cute_girl_mix4", "cuteGirlMix4_v10.safetensors")
            model = EditAnythingLoraModel(base_model_path=os.path.join(sd_models_path, "chilloutmix_NiPrunedFp32Fix"),
                                          lora_model_path=lora_model_path, use_blip=True, extra_inpaint=True,
                                          sam_generator=sam_generator,
                                          mask_predictor=mask_predictor,
                                          blip_processor=blip_processor,
                                          blip_model=blip_model,
                                          lora_weight=0.5,
                                          )
            create_demo_beauty(model.process, model.process_image_click)
        # with gr.TabItem(' 👨‍🌾Handsome Edit/Generation'):
        #     model = EditAnythingLoraModel(base_model_path=os.path.join(sd_models_path, "Realistic_Vision_V2.0"),
        #                                   lora_model_path=None, use_blip=True, extra_inpaint=True,
        #                                   sam_generator=sam_generator,
        #                                   mask_predictor=mask_predictor,
        #                                   blip_processor=blip_processor,
        #                                   blip_model=blip_model)
        #     create_demo_handsome(model.process, model.process_image_click)
        # with gr.TabItem('Edit More'):
        #     model = EditAnythingLoraModel(base_model_path="andite/anything-v4.0",
        #                                   lora_model_path=None, use_blip=True, extra_inpaint=True,
        #                                   sam_generator=sam_generator,
        #                                   mask_predictor=mask_predictor,
        #                                   blip_processor=blip_processor,
        #                                   blip_model=blip_model,
        #                                   lora_weight=0.5,
        #                                   )
            create_demo_beauty(model.process, model.process_image_click)
        # with gr.TabItem('Generate Anything'):
        #     create_demo_generate_anything()
    with gr.Tabs():
        gr.Markdown(SHARED_UI_WARNING)

demo.queue(api_open=False).launch(server_name='0.0.0.0', share=False)