Spaces:
Running
on
Zero
Running
on
Zero
Ahsen Khaliq
commited on
Commit
·
feaa03b
1
Parent(s):
7769341
Update app.py
Browse files
app.py
CHANGED
|
@@ -116,7 +116,7 @@ def proc_pil_img(input_image, model):
|
|
| 116 |
transformed_image = img_transforms(input_image)[None,...].cuda().half()
|
| 117 |
|
| 118 |
with torch.no_grad():
|
| 119 |
-
result_image = model(transformed_image)[0]
|
| 120 |
output_image = tensor2im(result_image)
|
| 121 |
output_image = output_image.detach().cpu().numpy().astype('uint8')
|
| 122 |
output_image = PIL.Image.fromarray(output_image)
|
|
|
|
| 116 |
transformed_image = img_transforms(input_image)[None,...].cuda().half()
|
| 117 |
|
| 118 |
with torch.no_grad():
|
| 119 |
+
result_image = model(transformed_image)[0]
|
| 120 |
output_image = tensor2im(result_image)
|
| 121 |
output_image = output_image.detach().cpu().numpy().astype('uint8')
|
| 122 |
output_image = PIL.Image.fromarray(output_image)
|