Update inference.py

inference.py

@@ -71,7 +71,6 @@ def segment_wt(wt_model, flair_tensor, t1_tensor, t1ce_tensor, t2_tensor):
         wt_output = wt_model(wt_input)
     return wt_output
 
-
 def segment_tc(tc_model, flair_tensor, t1_tensor, t1ce_tensor, t2_tensor, wt_output):
     tc_input = torch.cat((flair_tensor, t1_tensor, t1ce_tensor, t2_tensor, wt_output.squeeze(0), 
                         wt_output.squeeze(0), wt_output.squeeze(0), wt_output.squeeze(0)), dim=0).unsqueeze(0)