rm stuff

app.py

@@ -47,27 +47,27 @@ def generate_imgs_from_user(image,
         x = transforms.Grayscale(num_output_channels=1)(x)
         x = torch.cat((x, torch.zeros_like(x)), dim=1)
 
-    return generate_imgs(x, physics, use_gen, baseline, model, metrics)
+    return generate_imgs(x, physics, True, baseline, model, metrics)
 
 def generate_imgs_from_dataset(dataset: EvalDataset, idx: int,
-                               physics: PhysicsWithGenerator, use_gen: bool,
+                               physics: PhysicsWithGenerator, # use_gen: bool,
                                baseline: BaselineModel, model: EvalModel,
                                metrics: List[Metric]):
     ### Load 1 image
     x = dataset[idx]    # shape : (C, H, W)
     x = x.unsqueeze(0)  # shape : (1, C, H, W)
 
-    return generate_imgs(x, physics, use_gen, baseline, model, metrics)
+    return generate_imgs(x, physics, True, baseline, model, metrics)
 
 def generate_random_imgs_from_dataset(dataset: EvalDataset,
                                       physics: PhysicsWithGenerator,
-                                      use_gen: bool,
+                                      # use_gen: bool,
                                       baseline: BaselineModel,
                                       model: EvalModel,
                                       metrics: List[Metric]):
     idx = random.randint(0, len(dataset)-1)
     x, y, out, out_baseline, saved_params_str, metrics_y, metrics_out, metrics_out_baseline = generate_imgs_from_dataset(
-        dataset, idx, physics, use_gen, baseline, model, metrics
+        dataset, idx, physics, True, baseline, model, metrics
         )
     return idx, x, y, out, out_baseline, saved_params_str, metrics_y, metrics_out, metrics_out_baseline