Dockerfile yaml added

better-me-team · Jan 23, 2022 · eee4d8d · eee4d8d
1 parent c324cb9
commit eee4d8d
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 1 deletion.
diff --git a/.DS_Store b/.DS_Store
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,14 @@
+FROM python:3.8
+
+WORKDIR /better.me
+
+COPY requirements.txt ./requirements.txt
+
+RUN pip3 install -r requirements.txt
+
+EXPOSE 8080
+
+COPY . /better.me
+
+CMD streamlit run --server.port 8080 --server.enableCORS false app.py
+
diff --git a/app.yaml b/app.yaml
@@ -0,0 +1,4 @@
+runtime: custom
+env: flex
+manual_scaling:
+  instances: 1
diff --git a/model_loading.py b/model_loading.py
@@ -50,7 +50,7 @@ def configure_optimizers(self):
 
 def load_model(path):
     # return model
-    pass
+    return pt.load("./model.pt")
 
 
 def pred(text: str, model):

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,11 @@
+-f https://download.pytorch.org/whl/torch_stable.html
+torch==1.8.1+cpu
+pytorch_lightning
 streamlit~=1.4.0
 watchdog
 requests~=2.27.1
 pandas~=1.3.5
 numpy~=1.21.5
 altair~=4.2.0
+plotly
+transformers[torch]