Cognomen commited on
Commit
bca8e13
Β·
1 Parent(s): d652dd0

test without transforms

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -32,17 +32,17 @@ generator = torch.manual_seed(0)
32
  def infer(prompt, negative_prompt, image):
33
  # implement your inference function here
34
 
35
- cond_input = T.ToPILImage(conditioning_image_transforms(np.array(image)))
36
 
37
  output = pipe(
38
  prompt,
39
- cond_input,
40
  generator=generator,
41
- num_images_per_prompt=4,
42
  num_inference_steps=20
43
  )
44
 
45
- return output.images
46
 
47
  gr.Interface(
48
  infer,
 
32
  def infer(prompt, negative_prompt, image):
33
  # implement your inference function here
34
 
35
+ #cond_input = conditioning_image_transforms(np.array(image)))
36
 
37
  output = pipe(
38
  prompt,
39
+ image,
40
  generator=generator,
41
+ num_images_per_prompt=1,
42
  num_inference_steps=20
43
  )
44
 
45
+ return output.images[0]
46
 
47
  gr.Interface(
48
  infer,