Update app.py
Browse files
app.py
CHANGED
@@ -90,8 +90,8 @@ def process(
|
|
90 |
|
91 |
# if not mask:
|
92 |
# gr.Info("Please draw a mask on the image.")
|
93 |
-
|
94 |
-
|
95 |
prompt=input_text,
|
96 |
image=resized_image,
|
97 |
width=width,
|
@@ -101,7 +101,7 @@ def process(
|
|
101 |
joint_attention_kwargs={"scale": 1.2},
|
102 |
num_inference_steps=num_inference_steps_slider
|
103 |
).images[0]
|
104 |
-
|
105 |
return result, None
|
106 |
|
107 |
#resized_mask = mask.resize((width, height), Image.LANCZOS)
|
|
|
90 |
|
91 |
# if not mask:
|
92 |
# gr.Info("Please draw a mask on the image.")
|
93 |
+
pipe2.load_lora_weights("SIGMitch/KIT")
|
94 |
+
result = pipe2(
|
95 |
prompt=input_text,
|
96 |
image=resized_image,
|
97 |
width=width,
|
|
|
101 |
joint_attention_kwargs={"scale": 1.2},
|
102 |
num_inference_steps=num_inference_steps_slider
|
103 |
).images[0]
|
104 |
+
print('INFERENCE DONE')
|
105 |
return result, None
|
106 |
|
107 |
#resized_mask = mask.resize((width, height), Image.LANCZOS)
|