Spaces:
Runtime error
Runtime error
| # Edit Anything trained with Stable Diffusion + ControlNet + SAM + BLIP2 | |
| import os | |
| import gradio as gr | |
| from diffusers.utils import load_image | |
| from editany_lora import EditAnythingLoraModel, config_dict | |
| from editany_demo import create_demo_template | |
| from huggingface_hub import hf_hub_download, snapshot_download | |
| def create_demo(process, process_image_click=None): | |
| examples = None | |
| INFO = f""" | |
| ## EditAnything https://github.com/sail-sg/EditAnything | |
| """ | |
| WARNING_INFO = None | |
| demo = create_demo_template( | |
| process, | |
| process_image_click, | |
| examples=examples, | |
| INFO=INFO, | |
| WARNING_INFO=WARNING_INFO, | |
| enable_auto_prompt_default=True, | |
| ) | |
| return demo | |
| if __name__ == "__main__": | |
| 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, | |
| ) | |
| demo = create_demo(model.process, model.process_image_click) | |
| demo.queue().launch(server_name="0.0.0.0") | |