Update main.py

main.py

@@ -29,7 +29,7 @@ api_router = APIRouter()
 #     return {"message": "Hello World!"}
 
 
-@api_router.get("/classify/{image_name}", status_code=200, response_model=ClassificationResult)
+@api_router.get("/classify", status_code=200, response_model=ClassificationResult)
 async def predict_image(image_name, model=model):
     img = load_image(image_name)
     prediction, pred_idx, probs = model.predict(img)