feat: process image before converting it to 3d

app.py

@@ -299,7 +299,7 @@ import tempfile
 
 app = FastAPI()
 
-@app.post("/generatee")
+@app.post("/generate")
 async def generate_3d(image: UploadFile = File(...)):
     if not image:
         raise HTTPException(status_code=400, detail="No image provided")
@@ -307,7 +307,7 @@ async def generate_3d(image: UploadFile = File(...)):
     try:
         image_data = Image.open(image.file)
         image_data = image_data.convert("RGBA")
-        # image_data = torch.tensor(np.array(image_data)).to('cuda')
+        image_data = preprocess_image(image_data)
 
         seed = 42
         ss_guidance_strength = 7.5
@@ -341,9 +341,6 @@ async def generate_3d(image: UploadFile = File(...)):
         raise HTTPException(status_code=500, detail=str(e))
 
 
-@app.get("/test")
-async def test_():
-    return JSONResponse(content={"TEST": "DONE!"})
 @app.get("/")
 async def root_():
     return JSONResponse(content={"message": "HI!"})