Update app.py

app.py

@@ -70,8 +70,7 @@ def gen_pred(img, model=seg_model):
     img = tf.expand_dims(img, axis=0)
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
-    # fig = plt.figure(figsize=(10, 7))
-    fig = plt.figure()
+    fig = plt.figure(figsize=(4, 4))
     fig.add_subplot(rows, columns, 1)
     # plt.imshow(pred, interpolation='catrom')
     plt.imshow(pred)