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)
|