Upload inference.py

inference.py

@@ -38,6 +38,9 @@ def inference(pipe, fiber_imgs, ring_imgs, num_steps):
     # sizes
     tile = 512
     canvas_size = tile * 2 
+    
+    fiber_imgs = fiber_imgs.convert("L")
+    ring_imgs = ring_imgs.convert("L")
 
 
     # stack channels: [fiber, ring, ring] -> H,W,3