Spaces:
Running
on
L40S
Running
on
L40S
Commit
·
e4ec5a4
1
Parent(s):
4dedfc4
Update app.py
Browse files
app.py
CHANGED
@@ -388,7 +388,7 @@ def start(_, calibrate_prompts, user_id, request: gr.Request):
|
|
388 |
image = list(imageio.imiter(im))
|
389 |
image = image[len(image)//2]
|
390 |
|
391 |
-
im = torchvision.transforms.ToTensor()(
|
392 |
im = torch.nn.functional.interpolate(im, (224, 224))
|
393 |
im = (im - .5) * 2
|
394 |
|
|
|
388 |
image = list(imageio.imiter(im))
|
389 |
image = image[len(image)//2]
|
390 |
|
391 |
+
im = torchvision.transforms.ToTensor()(image).unsqueeze(0)
|
392 |
im = torch.nn.functional.interpolate(im, (224, 224))
|
393 |
im = (im - .5) * 2
|
394 |
|