Spaces:
Sleeping
Sleeping
typo
Browse files
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'
|
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()
|