Upload inference.py

inference.py

@@ -38,18 +38,13 @@ 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
     arr_f = np.array(fiber_imgs).astype(np.uint8)
     arr_r = np.array(ring_imgs).astype(np.uint8)
-    arr_in = np.stack([arr_f, arr_r, arr_r], axis=2)  # H,W,3
-    print(np.shape(arr_in))
-    print(np.shape(arr_f))
-    print(np.shape(arr_r))
+    arr_in = np.stack([arr_f[:,:,0], arr_r[:,:,0], arr_r[:,:,0]], axis=2)  # H,W,3
+
     input_image = PIL.Image.fromarray(arr_in)         # PIL RGB
 
     # run pipeline (use autocast consistent with device)