msong97 commited on
Commit
35c18b7
·
1 Parent(s): 6f0291c
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ def generate_imgs(x: torch.Tensor,
110
  out_baseline = out_baseline[..., w_1:w_2, h_1:h_2]
111
 
112
  ### Process y when y shape is different from x shape
113
- if physics.name == 'MRI' and physics.name == 'CT':
114
  y_plot = physics.physics.prox_l2(physics.physics.A_adjoint(y), y, 1e4)
115
  else:
116
  y_plot = y.clone()
 
110
  out_baseline = out_baseline[..., w_1:w_2, h_1:h_2]
111
 
112
  ### Process y when y shape is different from x shape
113
+ if physics.name == 'MRI' or physics.name == 'CT':
114
  y_plot = physics.physics.prox_l2(physics.physics.A_adjoint(y), y, 1e4)
115
  else:
116
  y_plot = y.clone()