nahidalam commited on
Commit
b2558f7
·
1 Parent(s): 9cde633

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -38,7 +38,8 @@ def run(image_path):
38
  '''
39
  #predict_and_save(img_array, model)
40
  img_array = tf.expand_dims(image_path, 0)
41
- predict_and_save(img_array, model)
 
42
 
43
  iface = gr.Interface(run, gr.inputs.Image(shape=(256, 256)), "image")
44
 
 
38
  '''
39
  #predict_and_save(img_array, model)
40
  img_array = tf.expand_dims(image_path, 0)
41
+ im = predict_and_save(img_array, model)
42
+ return im
43
 
44
  iface = gr.Interface(run, gr.inputs.Image(shape=(256, 256)), "image")
45