Hugging Face's logo

Spaces:
SBB
/
eynollah-demo
Running

App Files Community
vahidrezanezhad commited on
Commit
0b19517
·
verified ·
1 Parent(s): cdc00a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def visualize_model_output(prediction):
37
  output[:,:,1][prediction[:,:,0]==unq_class] = rgb_class_unique[1]
38
  output[:,:,2][prediction[:,:,0]==unq_class] = rgb_class_unique[2]
39
 
40
- return output
41
 
42
 
43
  def do_prediction(model_name, img):
 
37
  output[:,:,1][prediction[:,:,0]==unq_class] = rgb_class_unique[1]
38
  output[:,:,2][prediction[:,:,0]==unq_class] = rgb_class_unique[2]
39
 
40
+ return output.astype(np.uint8)
41
 
42
 
43
  def do_prediction(model_name, img):