Update model.py

model.py

@@ -83,7 +83,7 @@ class Model(LabelStudioMLBase):
                         logits = self.seg_model(**inputs).logits
                         logits = 1 / (1 + torch.exp(-logits))
                         print(logits)
-                        preds = logits > 0.25
+                        preds = logits > 0.5
                         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({