Update app.py

app.py

@@ -32,7 +32,7 @@ def create_image(cancer_type):
     else:
         diffusion_model_name = 'diffusion_dp'
     os.system('python pix2pixhd_test.py --name ' + diffusion_model_name + ' --dataroot ./tmp --label_nc 0 --results_dir ./tmp --no_instance --resize_or_crop none --checkpoints_dir ' + model_dir)
-    image_dir = "./tmp/diffusion_dp/test_latest/images"
+    image_dir = "./tmp/" + diffusion_model_name + "/test_latest/images"
     input_label_image = Image.open(os.path.join(image_dir, "sample_input_label.jpg"))
     synthesized_image = Image.open(os.path.join(image_dir, "sample_synthesized_image.jpg"))
     return input_label_image, synthesized_image