Update app.py

app.py

@@ -3,6 +3,7 @@ import transformers
 from torch import nn
 import numpy as np
 import gradio as gr
+import PIL
 
 # Instantiate classification model
 from fastai.vision.all import *
@@ -117,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)