specify cpu only build/index url

allenai · Jul 14, 2023 · 1ea5b64 · 1ea5b64
1 parent e2b7aac
commit 1ea5b64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/readme.md b/readme.md
@@ -1,6 +1,6 @@
 This repository contains a computer vision model along with a containerized restful API (FastAPI) for serving streaming detections of vessels in near real time. See [docs/model_card.md](./docs/model_card.md) for information about the model and [docs/openapi.json](./docs/openapi.json) for the API specification. This model was built for [Skylight](https://www.skylight.global/), a product of AI2 that supports maritime transparency through actionable intelligence in order to help protect our oceans.
 
-<p float="left">
+<p float="center">
   <img src="images/summary.png" width=700 >
 </p>
 

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cpu torch==2.0.1+cpu
 anyio==3.7.0
 appdirs==1.4.4
 attrs==23.1.0
@@ -80,7 +81,6 @@ sympy==1.11.1
 threadpoolctl==3.1.0
 tifffile==2023.4.12
 tomli==2.0.1
-torch==2.0.1+cpu
 torchvision==0.15.2
 typing_extensions==4.7.0
 urllib3<2.0.0