Update server.py

app/server.py

@@ -3,6 +3,11 @@ from fastapi import FastAPI
 from fastapi.responses import RedirectResponse
 from langserve import add_routes
 from app.chain import build_chain
+import mlflow
+
+
+mlflow.set_tracking_uri("http://localhost:5000")
+mlflow.langchain.autolog()
 
 # Input API
 class QueryInput(BaseModel):