Fix!

app.py

@@ -1,11 +1,11 @@
 from fastapi import FastAPI
 
-from routers.distilbert import predict
+from routers.distilbert import prediction
  
 # Create a new FastAPI app instance
 app = FastAPI()
 
-app.include_router(predict)
+app.include_router(prediction)
 
 @app.get("/")
 def greet_json():

routers/distilbert.py

@@ -3,12 +3,12 @@ from src.distilbert import Distilbert
  
 
 MODEL_PATH = "SebaK13/DistilBERT-finetuned-customer-queries-balanced"
-predict = APIRouter(prefix="/predict", tags=["predict"])
+prediction = APIRouter(prefix="/predict", tags=["predict"])
 distilbert_service = Distilbert(model_path=MODEL_PATH)
 
 
 
-@predict.get("/predict")
+@prediction.get("/predict")
 def predict(query: str):
     output = distilbert_service.predict_query_type(query)
     return {"label": output}