Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -426,7 +426,7 @@ height = 768
|
|
426 |
width = 768
|
427 |
###
|
428 |
global sd_model_path
|
429 |
-
sd_model_path = models_dict["
|
430 |
use_safetensors= False
|
431 |
### LOAD Stable Diffusion Pipeline
|
432 |
# pipe1 = StableDiffusionXLPipeline.from_pretrained(sd_model_path, torch_dtype=torch.float16, use_safetensors= use_safetensors)
|
@@ -648,7 +648,7 @@ with gr.Blocks(css=css) as demo:
|
|
648 |
# get_genprompt_run = gr.Button("(2) I've finished segment labeling ! 😺", elem_id="prompt_button", interactive=True)
|
649 |
|
650 |
with gr.Column(visible=True) as gen_prompt_vis:
|
651 |
-
sd_type = gr.Dropdown(choices=list(models_dict.keys()), value = "
|
652 |
model_type = gr.Radio(["Only Using Textual Description", "Using Ref Images"], label="model_type", value = "Only Using Textual Description", info="Control type of the Character")
|
653 |
with gr.Group(visible=False) as control_image_input:
|
654 |
files = gr.Files(
|
|
|
426 |
width = 768
|
427 |
###
|
428 |
global sd_model_path
|
429 |
+
sd_model_path = models_dict["SDXL"]#"SG161222/RealVisXL_V4.0"
|
430 |
use_safetensors= False
|
431 |
### LOAD Stable Diffusion Pipeline
|
432 |
# pipe1 = StableDiffusionXLPipeline.from_pretrained(sd_model_path, torch_dtype=torch.float16, use_safetensors= use_safetensors)
|
|
|
648 |
# get_genprompt_run = gr.Button("(2) I've finished segment labeling ! 😺", elem_id="prompt_button", interactive=True)
|
649 |
|
650 |
with gr.Column(visible=True) as gen_prompt_vis:
|
651 |
+
sd_type = gr.Dropdown(choices=list(models_dict.keys()), value = "SDXL",label="sd_type", info="Select pretrained model")
|
652 |
model_type = gr.Radio(["Only Using Textual Description", "Using Ref Images"], label="model_type", value = "Only Using Textual Description", info="Control type of the Character")
|
653 |
with gr.Group(visible=False) as control_image_input:
|
654 |
files = gr.Files(
|