yuragoithf commited on
Commit
7058c31
·
1 Parent(s): fda85b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def gen_pred(img, model=seg_model):
70
  img = tf.expand_dims(img, axis=0)
71
  pred = model.predict(img)
72
  pred = np.squeeze(pred, axis=0)
73
- fig = plt.figure(figsize=(2, 2))
74
  fig.add_subplot(rows, columns, 1)
75
  # plt.imshow(pred, interpolation='catrom')
76
  plt.imshow(pred)
 
70
  img = tf.expand_dims(img, axis=0)
71
  pred = model.predict(img)
72
  pred = np.squeeze(pred, axis=0)
73
+ fig = plt.figure(figsize=(3, 3))
74
  fig.add_subplot(rows, columns, 1)
75
  # plt.imshow(pred, interpolation='catrom')
76
  plt.imshow(pred)