John Smith commited on
Commit
c357975
·
1 Parent(s): b9cfa97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ def image_mod(image_mod):
12
  plt.imshow(resize.numpy().astype(int))
13
  yhat = model.predict(np.expand_dims(resize,0))
14
  display = np.max(yhat)
 
15
  return display
16
 
17
  gr.Interface(fn=image_mod,
 
12
  plt.imshow(resize.numpy().astype(int))
13
  yhat = model.predict(np.expand_dims(resize,0))
14
  display = np.max(yhat)
15
+ display = str(display)
16
  return display
17
 
18
  gr.Interface(fn=image_mod,