JoPmt commited on
Commit
ac4cdd6
1 Parent(s): bbd9368

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -18,11 +18,7 @@ def plex(prompt, fifth, twice):
18
  gc.collect()
19
  apol=[]
20
  prompt = prompt
21
- nm = random.randint(1, 4836928)
22
- while nm % 32 != 0:
23
- nm = random.randint(1, 4836928)
24
- generator = torch.Generator(device="cpu").manual_seed(nm)
25
- image = pipe([prompt]*fifth, generator=generator, num_inference_steps=twice)
26
  for a, imze in enumerate(image["images"]):
27
  apol.append(imze)
28
  return apol
 
18
  gc.collect()
19
  apol=[]
20
  prompt = prompt
21
+ image = pipe([prompt]*fifth, num_inference_steps=twice)
 
 
 
 
22
  for a, imze in enumerate(image["images"]):
23
  apol.append(imze)
24
  return apol