Update app.py

app.py

@@ -67,7 +67,7 @@ def gen_pred(img, model=seg_model):
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
     fig = plt.figure(figsize=(10, 7))
-    fig.add_subplot(rows, columns, 1)
+    fig.add_subplot(rows=1, columns=1, 1)
     plt.imshow(pred, interpolation='catrom')
     plt.axis('off')
     plt.title("Prediction")