Upload inference.py

inference.py

@@ -88,8 +88,11 @@ def inference(fiber_imgs, ring_imgs, num_steps):
     bl = ring_imgs.crop((0, tile, tile, canvas_size))
     br = ring_imgs.crop((tile, tile, canvas_size, canvas_size))
     
-    tr = cv2.resize(tr, (new_width, original_height), interpolation=cv2.INTER_LANCZOS4)
-    br = cv2.resize(br, (new_width, original_height), interpolation=cv2.INTER_LANCZOS4)
+    tr = cv2.resize(tr.astype(np.uint8), (new_width, original_height), interpolation=cv2.INTER_LANCZOS4)
+    br = cv2.resize(br.astype(np.uint8), (new_width, original_height), interpolation=cv2.INTER_LANCZOS4)
+    
+    tr = PIL.Image.fromarray(tr) 
+    br = PIL.Image.fromarray(br) 
 
     # close opened images to free handles
     fiber_imgs.close()