implement ssd-1b inference

aniketmaurya · Nov 30, 2023 · a1107ee · a1107ee
1 parent 3d3f0e4
commit a1107ee
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -16,16 +16,19 @@ pip install .
 python -m fastserve
 ```
 
-
 ## Usage/Examples
 
+### Serve SSD-1B
+
 ```python
-from fastserve import FastServe
+from fastserve.models import FastServeSSD
 
-serve = FastServe()
+serve = FastServeSSD(device="cuda")
 serve.run_server()
 ```
 
+or, run `python -m fastserve.models --model ssd-1b` from terminal.
+
 
 <!-- ## Demo