diff --git a/main.py b/main.py
index 8414f41..1636abb 100644
--- a/main.py
+++ b/main.py
@@ -357,6 +357,10 @@ def map_finish_reason(finish_reason: str) -> str:
 
 if __name__ == "__main__":
     import uvicorn
+    import asyncio
 
-    uvicorn.run("main:app", host="127.0.0.1", port=int(os.environ.get("PORT", "8000")),
+    try:
+        uvicorn.run("main:app", host="127.0.0.1", port=int(os.environ.get("PORT", "8000")),
                 log_level="debug" if debug else "critical", reload=debug, access_log=debug)
+    except (KeyboardInterrupt, asyncio.CancelledError):
+        pass