Upload 6 files

Dockerfile

@@ -0,0 +1,7 @@
+FROM python:3.8
+WORKDIR /app
+COPY requirements.txt ./requirements.txt
+RUN pip3 install -r requirements.txt
+EXPOSE 8080
+COPY . /app
+CMD streamlit run --server.port 8080 --server.enableCORS false app.py

app.py

@@ -0,0 +1,42 @@
+import streamlit as st
+from aitextgen import aitextgen
+from PIL import Image
+import pybase64
+
+
+def main():
+
+    def set_png_as_page_bg(png_file):
+        bin_str = get_base64_of_bin_file(png_file)
+        page_bg_img = '''
+        <style>
+        body {
+        background-image: url("data:image/png;base64,%s");
+        background-size: cover;
+        }
+        </style>
+        ''' % bin_str
+
+        st.markdown(page_bg_img, unsafe_allow_html=True)
+        return
+
+    @st.cache(allow_output_mutation=True)
+    def get_base64_of_bin_file(bin_file):
+        with open(bin_file, 'rb') as f:
+            data = f.read()
+        return pybase64.b64encode(data).decode()
+
+    ai = aitextgen('trained_model/pytorch_model.bin', config = 'trained_model/config.json')
+    set_png_as_page_bg('plato2.jpg')
+    st.header("What are you thinking?")
+    prompt_text = st.text_input(label = 'Ask Plato')
+    st.subheader("What do you think love is?")
+    st.subheader("Love is good. Love is fun and great. It is vast. It is vast. It is big. Love and life are realative but the truth is bitter."
+    "Sadly no one loves you. Some people don't get it.")
+    st.subheader(ai.generate_one(prompt = prompt_text, top_p = 0.9 ))
+
+
+
+
+if __name__ == '__main__':
+    main()

app.yaml

@@ -0,0 +1,11 @@
+runtime: custom
+env: flex
+
+entrypoint: find ${VIRTUAL_ENV}/lib/python3.8/site-packages/streamlit -type f \( -iname \*.py -o -iname \*.js \) -print0 | xargs -0 sed -i 's/healthz/health-check/g' && streamlit run app.py --server.port $PORT --server.enableCORS=false
+
+automatic_scaling:
+  min_num_instances : 1
+  max_num_instances : 7
+
+network:
+  session_affinity: true

requirements.txt

@@ -0,0 +1,7 @@
+streamlit==0.73.0
+pytorch-lightning==1.1.6
+torch==1.7.1
+torchvision==0.8.2
+aitextgen==0.3.0
+Pillow==7.2.0
+pybase64==1.1.3

trained_model/config.json

@@ -0,0 +1,31 @@
+{
+  "_name_or_path": "aitextgen/pytorch_model_124M.bin",
+  "activation_function": "gelu_new",
+  "architectures": [
+    "GPT2LMHeadModel"
+  ],
+  "attn_pdrop": 0.1,
+  "bos_token_id": 50256,
+  "embd_pdrop": 0.1,
+  "eos_token_id": 50256,
+  "gradient_checkpointing": false,
+  "initializer_range": 0.02,
+  "layer_norm_epsilon": 1e-05,
+  "model_type": "gpt2",
+  "n_ctx": 1024,
+  "n_embd": 768,
+  "n_head": 12,
+  "n_inner": null,
+  "n_layer": 12,
+  "n_positions": 1024,
+  "n_vocab": 50257,
+  "resid_pdrop": 0.1,
+  "summary_activation": null,
+  "summary_first_dropout": 0.1,
+  "summary_proj_to_labels": true,
+  "summary_type": "cls_index",
+  "summary_use_proj": true,
+  "transformers_version": "4.2.2",
+  "use_cache": true,
+  "vocab_size": 50257
+}

trained_model/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bee3ae3528f9a67d6c4688445f5c4bca194041f5a6d9471681ea3b3043f30a66
+size 510406247