nathanjc commited on
Commit
542be01
·
1 Parent(s): 32cfcb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -217,7 +217,7 @@ def get_output(input_image):
217
  with torch.no_grad():
218
  bbox,output_path = detect(opt)
219
  if os.path.exists(output_path):
220
- return np.array(Image.open(output_path))
221
  else:
222
  return input_image
223
 
 
217
  with torch.no_grad():
218
  bbox,output_path = detect(opt)
219
  if os.path.exists(output_path):
220
+ return Image.open(output_path)
221
  else:
222
  return input_image
223