Spaces:

Cognomen
/

CatCon-Controlnet-WD-1-5-b2

Runtime error

Cognomen commited on May 1, 2023

Commit

55bfa61

1 Parent(s): ce08539

enable augmentation

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,13 +33,14 @@ generator = torch.manual_seed(0)
 def infer(prompt, negative_prompt, image):
     # implement your inference function here
-    #cond_input = conditioning_image_transforms(np.array(image)))
     output = pipe(
         prompt,
-        image,
         generator=generator,
-        num_images_per_prompt=1,
         num_inference_steps=20
             )

 def infer(prompt, negative_prompt, image):
     # implement your inference function here
+    cond_input = conditioning_image_transforms(np.array(image)))
+    cond_input = T.ToPILImage(cond_input)
     output = pipe(
         prompt,
+        cond_input,
         generator=generator,
+        num_images_per_prompt=4,
         num_inference_steps=20
             )