Upload inference.py

inference.py

@@ -88,8 +88,8 @@ 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.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 = cv2.resize(np.asarray(tr), (new_width, original_height), interpolation=cv2.INTER_LANCZOS4)
+    br = cv2.resize(np.asarray(br), (new_width, original_height), interpolation=cv2.INTER_LANCZOS4)
     
     tr = PIL.Image.fromarray(tr) 
     br = PIL.Image.fromarray(br)