Cognomen commited on
Commit
e51ab9f
Β·
1 Parent(s): bfd5e5b

convert numpy input to tensor

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ generator = torch.manual_seed(0)
36
  def infer(prompt, negative_prompt, image):
37
  # implement your inference function here
38
 
39
- cond_input = conditioning_image_transforms(image)
40
  #cond_input = T.ToPILImage(cond_input)
41
 
42
  output = pipe(
 
36
  def infer(prompt, negative_prompt, image):
37
  # implement your inference function here
38
 
39
+ cond_input = conditioning_image_transforms(torch.from_numpy(image))
40
  #cond_input = T.ToPILImage(cond_input)
41
 
42
  output = pipe(