VIVEK JAYARAM commited on
Commit
adba228
·
1 Parent(s): 83bba75
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def process_image(image_choice, noise_sigma, operator_key, T, K):
43
  global model, ddim_scheduler, model_type
44
  if model is None:
45
  model_type = "diffusers"
46
- model_name = "google/ddpm-celebahq-256" if "Celeb" in image_choice else "google/ddpm-curch-256"
47
  model = DiffusionPipeline.from_pretrained(model_name).to(device).unet
48
  ddim_scheduler = DDIMScheduler(
49
  num_train_timesteps=1000,
 
43
  global model, ddim_scheduler, model_type
44
  if model is None:
45
  model_type = "diffusers"
46
+ model_name = "google/ddpm-celebahq-256" if "Celeb" in image_choice else "google/ddpm-church-256"
47
  model = DiffusionPipeline.from_pretrained(model_name).to(device).unet
48
  ddim_scheduler = DDIMScheduler(
49
  num_train_timesteps=1000,