Update app.py

app.py

@@ -70,7 +70,8 @@ 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(figsize=(10, 7))
+    fig = plt.figure()
     fig.add_subplot(rows, columns, 1)
     # plt.imshow(pred, interpolation='catrom')
     plt.imshow(pred)