tori29umai commited on
Commit
be7cf46
1 Parent(s): 8264f82
Files changed (2) hide show
  1. utils/prompt_analysis.py +4 -4
  2. utils/utils.py +1 -2
utils/prompt_analysis.py CHANGED
@@ -15,14 +15,14 @@ class PromptAnalysis:
15
  self.model = None
16
  self.model_dir = model_dir
17
 
18
- def layout(self, lang_util, input_image):
19
  with gr.Column():
20
  with gr.Row():
21
- self.prompt = gr.Textbox(label=lang_util.get_text("prompt"), lines=3)
22
  with gr.Row():
23
- self.negative_prompt = gr.Textbox(label=lang_util.get_text("negative_prompt"), lines=3, value=self.default_nagative_prompt)
24
  with gr.Row():
25
- self.prompt_analysis_button = gr.Button(lang_util.get_text("analyze_prompt"))
26
 
27
  self.prompt_analysis_button.click(
28
  self.process_prompt_analysis,
 
15
  self.model = None
16
  self.model_dir = model_dir
17
 
18
+ def layout(self, input_image):
19
  with gr.Column():
20
  with gr.Row():
21
+ self.prompt = gr.Textbox("prompt", lines=3)
22
  with gr.Row():
23
+ self.negative_prompt = gr.Textbox("negative_prompt", lines=3, value=self.default_nagative_prompt)
24
  with gr.Row():
25
+ self.prompt_analysis_button = gr.Button("analyze_prompt")
26
 
27
  self.prompt_analysis_button.click(
28
  self.process_prompt_analysis,
utils/utils.py CHANGED
@@ -11,8 +11,7 @@ import cv2
11
  def load_cn_model(model_dir):
12
  folder = model_dir
13
  file_name = 'diffusion_pytorch_model.safetensors'
14
- url = "https://huggingface.co/kataragi/ControlNet-LineartXL/resolve/main/Katarag_lineartXL-fp16.safetensors"
15
-
16
  file_path = os.path.join(folder, file_name)
17
  if not os.path.exists(file_path):
18
  response = requests.get(url, stream=True)
 
11
  def load_cn_model(model_dir):
12
  folder = model_dir
13
  file_name = 'diffusion_pytorch_model.safetensors'
14
+ url = " https://huggingface.co/2vXpSwA7/iroiro-lora/resolve/main/test_controlnet2/CN-anytest_v3-50000_fp16.safetensors"
 
15
  file_path = os.path.join(folder, file_name)
16
  if not os.path.exists(file_path):
17
  response = requests.get(url, stream=True)