Update app.py
Browse files
app.py
CHANGED
@@ -74,7 +74,7 @@ logger.info("Pipeline loaded successfully.")
|
|
74 |
MAX_SEED = 1000000
|
75 |
MAX_PIXEL_BUDGET = 1024 * 1024
|
76 |
|
77 |
-
|
78 |
def process_input(input_image, upscale_factor):
|
79 |
w, h = input_image.size
|
80 |
aspect_ratio = w / h
|
@@ -98,7 +98,7 @@ def process_input(input_image, upscale_factor):
|
|
98 |
|
99 |
return input_image.resize((w, h)), was_resized
|
100 |
|
101 |
-
|
102 |
def run_inference(input_image, upscale_factor, seed, num_inference_steps, controlnet_conditioning_scale):
|
103 |
logger.info("Processing inference.")
|
104 |
input_image, was_resized = process_input(input_image, upscale_factor)
|
|
|
74 |
MAX_SEED = 1000000
|
75 |
MAX_PIXEL_BUDGET = 1024 * 1024
|
76 |
|
77 |
+
#@spaces.GPU
|
78 |
def process_input(input_image, upscale_factor):
|
79 |
w, h = input_image.size
|
80 |
aspect_ratio = w / h
|
|
|
98 |
|
99 |
return input_image.resize((w, h)), was_resized
|
100 |
|
101 |
+
#@spaces.GPU
|
102 |
def run_inference(input_image, upscale_factor, seed, num_inference_steps, controlnet_conditioning_scale):
|
103 |
logger.info("Processing inference.")
|
104 |
input_image, was_resized = process_input(input_image, upscale_factor)
|