vahidrezanezhad commited on
Commit
a0c54ca
1 Parent(s): 48a54dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def do_prediction(model_name, img):
136
  '''
137
  #prediction_true = prediction_true * -1
138
  #prediction_true = prediction_true + 1
139
- return "No numerical output", prediction_true[:,:,0]# * 255
140
 
141
  # catch-all (we should not reach this)
142
  case _:
 
136
  '''
137
  #prediction_true = prediction_true * -1
138
  #prediction_true = prediction_true + 1
139
+ return "No numerical output", prediction_true
140
 
141
  # catch-all (we should not reach this)
142
  case _: