add getters

app.py

@@ -4,6 +4,16 @@ from fastapi.responses import HTMLResponse
 
 app = FastAPI()
 
+def get_x(i):
+    # Convert NumPy array to a single-channel PIL image with inverted colors
+    return PILImageBW.create(all_noise[i])
+
+def get_y(i):
+    return all_thresh[i].astype(np.float32)
+
+def get_items(_):
+    return range(len(all_noise))
+
 # Load the model
 learn = load_learner('model.pkl')
 
@@ -24,15 +34,7 @@ if __name__ == "__main__":
 # import base64
 # import uvicorn
 
-# def get_x(i):
-#     # Convert NumPy array to a single-channel PIL image with inverted colors
-#     return PILImageBW.create(all_noise[i])
-
-# def get_y(i):
-#     return all_thresh[i].astype(np.float32)
 
-# def get_items(_):
-#     return range(len(all_noise))
 
 # # Load the model
 # learn = load_learner('model.pkl')