rphrp1985 commited on
Commit
bf62e77
1 Parent(s): 035676f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -28,7 +28,14 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
28
  seed = random.randint(0, MAX_SEED)
29
 
30
  generator = torch.Generator().manual_seed(seed)
31
- image = pipeline(prompt).images[0]
 
 
 
 
 
 
 
32
 
33
 
34
 
 
28
  seed = random.randint(0, MAX_SEED)
29
 
30
  generator = torch.Generator().manual_seed(seed)
31
+ image = pipeline(prompt = prompt,
32
+ negative_prompt = negative_prompt,
33
+ guidance_scale = guidance_scale,
34
+ num_inference_steps = num_inference_steps,
35
+ width = width,
36
+ height = height,
37
+ generator = generator
38
+ ).images[0]
39
 
40
 
41