Spaces:
Running
on
Zero
Running
on
Zero
File size: 587 Bytes
41978ed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from typing import Any, Dict
import gradio as gr
from core.pipelines.sd_image_pipeline import SdImagePipeline
from core.pipelines.newbie_xml_pipeline import NewbieXmlPipeline
sd_image_pipeline = SdImagePipeline()
newbie_xml_pipeline = NewbieXmlPipeline()
def generate_image_wrapper(ui_inputs: dict, progress=gr.Progress(track_tqdm=True)):
return sd_image_pipeline.run(ui_inputs=ui_inputs, progress=progress)
def generate_xml_wrapper(ui_inputs: dict, progress=gr.Progress(track_tqdm=True)):
return newbie_xml_pipeline.run(ui_inputs=ui_inputs, progress=progress) |