Update model.py

model.py

@@ -81,9 +81,9 @@ class Model(LabelStudioMLBase):
                         image = image.crop((x.item(), y.item(), x2.item(), y2.item()))
                         inputs = self.seg_image_processor(images=image, return_tensors="pt")
                         logits = self.seg_model(**inputs).logits
-                        logits = torch.exp(logits)
+                        logits = 1 / (1 + torch.exp(-logits))
                         print(logits)
-                        preds = logits > 0.5
+                        preds = logits > 0.25
                         preds = [self.seg_id2label[i] for i, pred in enumerate(preds.squeeze().tolist()) if pred]
                         preds = ["No Reportado"] if "No Reportado" in preds else preds
                         result_list.append({