Cognomen commited on
Commit
cf1b1ed
Β·
1 Parent(s): fad62be

use fromarray

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,9 +36,9 @@ generator = torch.manual_seed(0)
36
  # inference function takes prompt, negative prompt and image
37
  def infer(prompt, negative_prompt, image):
38
  # implement your inference function here
39
- inp = PIL.Image.create(image)
40
 
41
- cond_input = conditioning_image_transforms(image)
42
  #cond_input = T.ToPILImage(cond_input)
43
 
44
  output = pipe(
 
36
  # inference function takes prompt, negative prompt and image
37
  def infer(prompt, negative_prompt, image):
38
  # implement your inference function here
39
+ inp = PIL.Image.fromarray(image)
40
 
41
+ cond_input = conditioning_image_transforms(inp)
42
  #cond_input = T.ToPILImage(cond_input)
43
 
44
  output = pipe(