Update app.py

app.py

@@ -113,8 +113,8 @@ if image_file is not None:
      
     img_cv=np.reshape(img_cv,(1,512,512,1))
     predict_img=model.predict(img_cv)
-    predict=predict_img[1,:,:,0]
-    plt.imsave("/content/predict.png",predict)
+#    predict=predict_img[1,:,:,0]
+    plt.imsave("/content/predict.png",predict_img)
     
     ## Plotting - Пример результата
     img = cv2.imread(image_file)