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