Update worker.py
Browse files
worker.py
CHANGED
|
@@ -349,7 +349,7 @@ def closestNumber(n, m):
|
|
| 349 |
|
| 350 |
@torch.inference_mode()
|
| 351 |
def generate(input):
|
| 352 |
-
values =
|
| 353 |
input_fg = values['input_fg']
|
| 354 |
input_fg = load_image(input_fg)
|
| 355 |
input_fg = np.asarray(input_fg)
|
|
|
|
| 349 |
|
| 350 |
@torch.inference_mode()
|
| 351 |
def generate(input):
|
| 352 |
+
values = input["input"]
|
| 353 |
input_fg = values['input_fg']
|
| 354 |
input_fg = load_image(input_fg)
|
| 355 |
input_fg = np.asarray(input_fg)
|