osanseviero commited on
Commit
912f1cb
1 Parent(s): 2187818

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
  pipeline = LatentDiffusionUncondPipeline.from_pretrained("CompVis/latent-diffusion-celeba-256")
8
 
9
 
10
- def predict(steps=1, seed):
11
  generator = torch.manual_seed(seed)
12
  image = pipeline(generator=generator, num_inference_steps=steps)["sample"]
13
  image_processed = image.cpu().permute(0, 2, 3, 1)
 
7
  pipeline = LatentDiffusionUncondPipeline.from_pretrained("CompVis/latent-diffusion-celeba-256")
8
 
9
 
10
+ def predict(steps=1, seed=42):
11
  generator = torch.manual_seed(seed)
12
  image = pipeline(generator=generator, num_inference_steps=steps)["sample"]
13
  image_processed = image.cpu().permute(0, 2, 3, 1)