Ahsen Khaliq commited on
Commit
feaa03b
1 Parent(s): 7769341

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]; print(result_image.shape)
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)