VIVEK JAYARAM
commited on
Commit
·
adba228
1
Parent(s):
83bba75
bug fix
Browse files
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-
|
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,
|