Branch with API files ready to go

CederGroupHub · May 13, 2021 · e1bcc05 · e1bcc05
1 parent 0efad16
commit e1bcc05
Show file tree

Hide file tree

Showing 10 changed files with 19 additions and 234 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,6 +1,21 @@
-FROM tiangolo/uvicorn-gunicorn-fastapi:python3.7
+FROM python:3.6.12
 
-COPY . /app
-COPY ./examples/predict_fastapi.py /app/main.py
-RUN pip install -e /app
+WORKDIR /classification
+
+# Install transformers
+RUN pip3 install transformers
+# Install fastAPI
+RUN pip3 install -U fastapi[all]
+# Install torch
+RUN pip3 install torch
+# Install mongo client
+RUN pip3 install pymongo
+# Install numpy/scipy
+RUN pip3 install scipy
+RUN pip3 install numpy
+
+COPY . /classification
+RUN pip3 install -e .
 RUN python -m synthesis_classifier.model download
+
+#uvicorn classification_api:app --port 8051 --reload --host 0.0.0.0
diff --git a/examples/predict_fastapi.py → classification_api.py b/examples/predict_fastapi.py → classification_api.py
diff --git a/examples/examples.txt b/examples/examples.txt
diff --git a/examples/predict_json.py b/examples/predict_json.py
diff --git a/synpro_scripts/classify_experimental.py b/synpro_scripts/classify_experimental.py
diff --git a/synpro_scripts/compute_embeddings.py b/synpro_scripts/compute_embeddings.py
diff --git a/synpro_scripts/convert_synpro.js b/synpro_scripts/convert_synpro.js
diff --git a/synpro_scripts/patents_batteries.py b/synpro_scripts/patents_batteries.py
diff --git a/synpro_scripts/patents_examples.py b/synpro_scripts/patents_examples.py
diff --git a/synpro_scripts/reclassify.py b/synpro_scripts/reclassify.py