huzefa11 commited on
Commit
91f8d5b
·
verified ·
1 Parent(s): f0953a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -426,7 +426,7 @@ height = 768
426
  width = 768
427
  ###
428
  global sd_model_path
429
- sd_model_path = models_dict["Unstable"]#"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,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 = "Unstable",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(
 
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(