no middleware

main.py

@@ -61,7 +61,7 @@ class ErrorOutput(BaseModel):
 
 @app.post('/generate', response_model=Union[GenerationOutput, ErrorOutput])
 def generate(request: GenerationRequest):
-    
+    t = time.perf_counter()
     print(request)
     try:
         output = sanitize_input(request)
@@ -72,6 +72,8 @@ def generate(request: GenerationRequest):
             return {'error': output[1]}
         
         images = model.sample(request)
+        process_time = time.perf_counter() - t
+        logger.info(f"Request took {process_time:0.3f} seconds")
         return {"generations": images}
 
     except Exception as e:
@@ -85,22 +87,5 @@ def generate(request: GenerationRequest):
             os.kill(os.getpid(), signal.SIGTERM)
         return {"error": str(e)}
 
-@app.middleware("http")
-async def handle_logging_and_errors(request: Request, call_next):
-    t = time.perf_counter()
-    response = await call_next(request)
-    process_time = time.perf_counter() - t
-    response.headers["X-Process-Time"] = str(process_time)
-    logger.info(f"Request took {t:0.3f} seconds")
-
-    f = open("/tmp/health_readiness", "w")
-    f.close()
-
-    if os.environ['DEV'] == "False":
-        f = open("/tmp/healthy", "w")
-        f.close()
-
-    return response
-
 if __name__ == "__main__":
     uvicorn.run("main:app", host="0.0.0.0", port=80, log_level="info")
\ No newline at end of file