lixiang46 commited on
Commit
1ae1c49
β€’
1 Parent(s): 0d5da93
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -44,8 +44,7 @@ def infer(prompt,
44
  if randomize_seed:
45
  seed = random.randint(0, MAX_SEED)
46
  generator = torch.Generator().manual_seed(seed)
47
- pil_image = Image.open(image)
48
- width, height = pil_image.size
49
  result = pipe(
50
  prompt = prompt,
51
  image = image['background'],
 
44
  if randomize_seed:
45
  seed = random.randint(0, MAX_SEED)
46
  generator = torch.Generator().manual_seed(seed)
47
+ width, height = image['background'].size
 
48
  result = pipe(
49
  prompt = prompt,
50
  image = image['background'],