Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
b3b6580
1
Parent(s):
96ff376
app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def load_model(lora_dir, cn_dir):
|
|
| 37 |
pipe.load_lora_weights(lora_dir, weight_name="lineart.safetensors")
|
| 38 |
return pipe
|
| 39 |
|
| 40 |
-
@spaces.GPU
|
| 41 |
def predict(input_image_path, prompt, negative_prompt, controlnet_scale):
|
| 42 |
pipe = load_model(lora_dir, cn_dir)
|
| 43 |
input_image = Image.open(input_image_path)
|
|
|
|
| 37 |
pipe.load_lora_weights(lora_dir, weight_name="lineart.safetensors")
|
| 38 |
return pipe
|
| 39 |
|
| 40 |
+
@spaces.GPU(duration=120)
|
| 41 |
def predict(input_image_path, prompt, negative_prompt, controlnet_scale):
|
| 42 |
pipe = load_model(lora_dir, cn_dir)
|
| 43 |
input_image = Image.open(input_image_path)
|