srijaydeshpande commited on
Commit
174bb9e
·
verified ·
1 Parent(s): cc26920

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def create_image(cancer_type):
29
  os.makedirs(tmp_dir)
30
  generate_latent(model_dir, cancer_type, tmp_dir)
31
  create_mask(model_dir, "./tmp", "./tmp/test_masks")
32
- os.system('python pix2pixhd_test.py --name diffusion_dp --dataroot ./tmp --label_nc 0 --no_instance --resize_or_crop none --checkpoints_dir ' + model_dir)
33
  image_dir = "./tmp/diffusion_dp/test_latest/images"
34
  input_label_image = Image.open(os.path.join(image_dir, "sample_input_label.jpg"))
35
  synthesized_image = Image.open(os.path.join(image_dir, "sample_synthesized_image.jpg"))
 
29
  os.makedirs(tmp_dir)
30
  generate_latent(model_dir, cancer_type, tmp_dir)
31
  create_mask(model_dir, "./tmp", "./tmp/test_masks")
32
+ os.system('python pix2pixhd_test.py --name diffusion_dp --dataroot ./tmp --label_nc 0 --results_dir ./tmp --no_instance --resize_or_crop none --checkpoints_dir ' + model_dir)
33
  image_dir = "./tmp/diffusion_dp/test_latest/images"
34
  input_label_image = Image.open(os.path.join(image_dir, "sample_input_label.jpg"))
35
  synthesized_image = Image.open(os.path.join(image_dir, "sample_synthesized_image.jpg"))