Make osam serve optional

README.md

@@ -30,6 +30,12 @@
 pip install osam
 ```
 
+**For `osam serve`**:
+
+```bash
+pip install osam[serve]
+```
+
 ## Quickstart
 
 To run with EfficientSAM:
@@ -102,6 +108,8 @@ PIL.Image.fromarray(response.mask).save("mask.png")
 ### HTTP
 
 ```bash
+# pip install osam[serve]  # required for `osam serve`
+
 # Get up the server
 osam serve
 

osam/__main__.py

@@ -8,7 +8,6 @@
 import imgviz
 import numpy as np
 import PIL.Image
-import uvicorn
 from loguru import logger
 
 from . import __version__
@@ -97,8 +96,15 @@ def rm(model_name):
 @cli.command(help="Start server")
 @click.option("--reload", is_flag=True, help="reload server on file changes")
 def serve(reload):
-    logger.info("Starting server...")
-    uvicorn.run("osam._server:app", host="127.0.0.1", port=11368, reload=reload)
+    try:
+        import uvicorn
+        import osam._server
+
+        logger.info("Starting server...")
+        uvicorn.run("osam._server:app", host="127.0.0.1", port=11368, reload=reload)
+    except ImportError:
+        logger.error("Run `pip install osam[serve]` to use `osam serve`")
+        sys.exit(1)
 
 
 @cli.command(help="Run a model")

pyproject.toml

@@ -32,17 +32,19 @@ classifiers = [
 dependencies = [
   "click",
   "gdown",
-  "fastapi",
   "imgviz",
   "loguru",
   "onnxruntime",
   "Pillow",
   "pydantic",
-  "uvicorn",
 ]
 dynamic = ["readme", "version"]
 
 [project.optional-dependencies]
+serve = [
+  "fastapi",
+  "uvicorn",
+]
 test = [
   "build",
   "mypy",