adpro commited on
Commit
f85a6e2
·
verified ·
1 Parent(s): 7cdadd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -27,8 +27,9 @@ def process_image(image):
27
  ).squeeze()
28
  output = prediction.cpu().numpy()
29
  formatted = (output * 255 / np.max(output)).astype('uint8')
 
30
  img = Image.fromarray(formatted)
31
- return img
32
 
33
  return result
34
 
 
27
  ).squeeze()
28
  output = prediction.cpu().numpy()
29
  formatted = (output * 255 / np.max(output)).astype('uint8')
30
+
31
  img = Image.fromarray(formatted)
32
+ return [img, gltf_path, gltf_path]
33
 
34
  return result
35