Update app.py

app.py

@@ -120,7 +120,7 @@ def predict(classification_mode, image):
 
         seg_img = PIL.Image.fromarray(seg_img)
 
-  return ({labels[i]: float(probs[i]) for i in range(len(labels))}, seg_img, percentage_affected)
+    return ({labels[i]: float(probs[i]) for i in range(len(labels))}, seg_img, percentage_affected)
 
 
 description = """