Spaces:
Sleeping
Sleeping
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(
|