Added lsun church

app.py

@@ -43,6 +43,7 @@ def process_image(image_choice, noise_sigma, operator_key, T, K):
     global model, ddim_scheduler, model_type
     if model is None:
         model_type = "diffusers"
+        model_name = "google/ddpm-celebahq-256" if "Celeb" in image_choice else "google/ddpm-curch-256"
         model = DiffusionPipeline.from_pretrained("google/ddpm-celebahq-256").to(device).unet
         ddim_scheduler = DDIMScheduler(
             num_train_timesteps=1000,
@@ -54,7 +55,8 @@ def process_image(image_choice, noise_sigma, operator_key, T, K):
     image_paths = {
         "CelebA HQ 1": "sample_images/celebhq_29999.jpg",
         "CelebA HQ 2": "sample_images/celebhq_00001.jpg",
-        "CelebA HQ 3": "sample_images/celebhq_00000.jpg"
+        "CelebA HQ 3": "sample_images/celebhq_00000.jpg",
+        "LSUN Church": "sample_images/lsun_church.png"
     }
 
     config_paths = {
@@ -99,7 +101,7 @@ with gr.Blocks() as demo:
         noise_sigma = gr.Slider(0, 0.6, value=0.05, step=0.01, label="Noise Sigma")
     
     image_select = gr.Dropdown(
-        choices=["CelebA HQ 1", "CelebA HQ 2", "CelebA HQ 3"],
+        choices=["CelebA HQ 1", "CelebA HQ 2", "CelebA HQ 3", "LSUN Church"],
         value="CelebA HQ 1",
         label="Select Input Image"
     )