Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -136,9 +136,7 @@ ratio_list = [[800, 800], [768, 896], [704, 928], [672, 960], [640, 1024], [608,
|
|
| 136 |
ratio_list += [[896, 768], [928, 704], [960, 672], [1024, 640], [1056, 608], [1088, 576], [1184, 576]]
|
| 137 |
|
| 138 |
def get_rate(image):
|
| 139 |
-
# 计算输入图片的长宽比
|
| 140 |
input_rate = image.size[0] / image.size[1]
|
| 141 |
-
# 计算与每个预设比例的差距
|
| 142 |
min_diff = float('inf')
|
| 143 |
best_idx = 0
|
| 144 |
|
|
@@ -292,7 +290,9 @@ def load_ckpt():
|
|
| 292 |
)
|
| 293 |
|
| 294 |
pipeline = pipeline.to("cuda")
|
| 295 |
-
|
|
|
|
|
|
|
| 296 |
global cur_style
|
| 297 |
cur_style = 'line + shadow'
|
| 298 |
@spaces.GPU
|
|
@@ -333,9 +333,7 @@ def change_ckpt(style):
|
|
| 333 |
|
| 334 |
return style
|
| 335 |
|
| 336 |
-
|
| 337 |
-
|
| 338 |
-
load_ckpt()
|
| 339 |
|
| 340 |
@spaces.GPU
|
| 341 |
def fix_random_seeds(seed):
|
|
|
|
| 136 |
ratio_list += [[896, 768], [928, 704], [960, 672], [1024, 640], [1056, 608], [1088, 576], [1184, 576]]
|
| 137 |
|
| 138 |
def get_rate(image):
|
|
|
|
| 139 |
input_rate = image.size[0] / image.size[1]
|
|
|
|
| 140 |
min_diff = float('inf')
|
| 141 |
best_idx = 0
|
| 142 |
|
|
|
|
| 290 |
)
|
| 291 |
|
| 292 |
pipeline = pipeline.to("cuda")
|
| 293 |
+
|
| 294 |
+
load_ckpt()
|
| 295 |
+
|
| 296 |
global cur_style
|
| 297 |
cur_style = 'line + shadow'
|
| 298 |
@spaces.GPU
|
|
|
|
| 333 |
|
| 334 |
return style
|
| 335 |
|
| 336 |
+
|
|
|
|
|
|
|
| 337 |
|
| 338 |
@spaces.GPU
|
| 339 |
def fix_random_seeds(seed):
|