randomtable commited on
Commit
7c05f45
1 Parent(s): 63ad20a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ def predict(
93
  pred = np.repeat(np.expand_dims(pred, axis=-1), 3, axis=-1)
94
  image_pred = (pred * np.array(image_pil)).astype(np.uint8)
95
 
96
- return image, image_pred
97
 
98
 
99
  examples = [[_] for _ in glob('examples/*')][:]
 
93
  pred = np.repeat(np.expand_dims(pred, axis=-1), 3, axis=-1)
94
  image_pred = (pred * np.array(image_pil)).astype(np.uint8)
95
 
96
+ return image_pred
97
 
98
 
99
  examples = [[_] for _ in glob('examples/*')][:]