Spaces:
Running
on
T4
Running
on
T4
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)
|