Spaces:
Sleeping
Sleeping
Bobby
commited on
Commit
·
25410b7
1
Parent(s):
74542b8
another laod normalbae
Browse files
app.py
CHANGED
|
@@ -406,6 +406,7 @@ def process_image(
|
|
| 406 |
|
| 407 |
seed = random.randint(0, MAX_SEED)
|
| 408 |
generator = torch.cuda.manual_seed(seed)
|
|
|
|
| 409 |
control_image = preprocessor(
|
| 410 |
image=image,
|
| 411 |
image_resolution=image_resolution,
|
|
|
|
| 406 |
|
| 407 |
seed = random.randint(0, MAX_SEED)
|
| 408 |
generator = torch.cuda.manual_seed(seed)
|
| 409 |
+
preprocessor.load("NormalBae")
|
| 410 |
control_image = preprocessor(
|
| 411 |
image=image,
|
| 412 |
image_resolution=image_resolution,
|