|
|
|
|
|
import os |
|
|
|
import gradio as gr |
|
import torch |
|
|
|
from app_base import create_demo as create_demo_base |
|
from app_sketch import create_demo as create_demo_sketch |
|
from model import ADAPTER_NAMES, Model, download_all_adapters |
|
|
|
DESCRIPTION = "# EG - ArtStudio" |
|
|
|
download_all_adapters() |
|
model = Model(ADAPTER_NAMES[0]) |
|
|
|
with gr.Blocks(css="style.css") as demo: |
|
gr.Markdown(DESCRIPTION) |
|
with gr.Tabs(): |
|
with gr.Tab(label="Base"): |
|
create_demo_base(model) |
|
with gr.Tab(label="Sketch"): |
|
create_demo_sketch(model) |
|
|
|
if __name__ == "__main__": |
|
demo.queue(max_size=20).launch(share=True) |