Commit
·
7058c31
1
Parent(s):
fda85b0
Update app.py
Browse files
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=(
|
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)
|