Update app.py

app.py

@@ -115,7 +115,7 @@ 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_img)
+    plt.imsave("predict.png",predict_img)
     
     ## Plotting - Пример результата
     img = cv2.imread(image_file)