change model to attention unet

Pixilib · Dec 27, 2023 · 9961f6a · 9961f6a
1 parent 2f921b1
commit 9961f6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php b/GaelO2/app/GaelO/Services/GaelOProcessingService/TmtvProcessingService.php
@@ -46,7 +46,7 @@ public function runInference(): MaskProcessingService
             'idCT' => $idCT
         ];
 
-        $inferenceResponse = $this->gaelOProcessingService->executeInference('unet_model', $inferencePayload);
+        $inferenceResponse = $this->gaelOProcessingService->executeInference('pt_seg_attentionunet', $inferencePayload);
         $maskId = $inferenceResponse['id_mask'];
         $maskProcessingService = new MaskProcessingService($this->orthancService, $this->gaelOProcessingService);
         $maskProcessingService->setMaskId($maskId);