Update model.py

model.py

@@ -76,23 +76,25 @@ class Model(LabelStudioMLBase):
                 x, y, x2, y2 = tuple(box)
 
                 if self.id2label[label.item()] == 'Propuesta':
-                    pred_label_id = str(uuid4())
-                    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)
-                    preds = logits > 0.5
-                    preds = [self.seg_id2label[i] for i, pred in enumerate(preds)]
-                    preds = ["No Reportado"] if "No Reportado" in preds else preds
-                    result_list.append({
-                      "value": {
-                        "choices": preds
-                      },
-                      "id": pred_label_id,
-                      "from_name": "propuesta",
-                      "to_name": "image",
-                      "type": "choices"
-                    })
+                    with torch.no_grad():
+                        pred_label_id = str(uuid4())
+                        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)
+                        print(logits)
+                        preds = logits > 0.5
+                        preds = [self.seg_id2label[i] for i, pred in enumerate(preds)]
+                        preds = ["No Reportado"] if "No Reportado" in preds else preds
+                        result_list.append({
+                          "value": {
+                            "choices": preds
+                          },
+                          "id": pred_label_id,
+                          "from_name": "propuesta",
+                          "to_name": "image",
+                          "type": "choices"
+                        })
                 
                 result_list.append({
                     'id': label_id,