Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Commit
•
be7cf46
1
Parent(s):
8264f82
app.py
Browse files- utils/prompt_analysis.py +4 -4
- 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,
|
19 |
with gr.Column():
|
20 |
with gr.Row():
|
21 |
-
self.prompt = gr.Textbox(
|
22 |
with gr.Row():
|
23 |
-
self.negative_prompt = gr.Textbox(
|
24 |
with gr.Row():
|
25 |
-
self.prompt_analysis_button = gr.Button(
|
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/
|
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)
|