Update app.py

app.py

@@ -118,7 +118,7 @@ def predict(classification_mode, image):
         percentage_affected = round((pixels_count[1]/960)*100, 1)
         percentage_affected = str(percentage_affected) + '%'
 
-        seg_img = PIL.Image.fromarray(seg_img)
+        #seg_img = PIL.Image.fromarray(seg_img)
 
     return ({labels[i]: float(probs[i]) for i in range(len(labels))}, seg_img, percentage_affected)