Update app.py

app.py

@@ -115,7 +115,8 @@ def predict(classification_mode, image):
         classified_pixels = np.unique(pred_seg.numpy(), return_counts=True)
         pixels_count = dict({classified_pixels[0][0]: classified_pixels[1][0],
                              classified_pixels[0][1]: classified_pixels[1][1]})
-        percentage_affected = round((pixels_count[1]/960)*100, 1)
+        #percentage_affected = round((pixels_count[1]/960)*100, 1)
+        percentage_affected = round((pixels_count[1]/(pixels_count[0]+pixels_count[1]))*100, 1)
         percentage_affected = str(percentage_affected) + '%'
 
         #seg_img = PIL.Image.fromarray(seg_img)