Duplicate from projects-jeru/container-test

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.*/

Dockerfile

@@ -0,0 +1,21 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+RUN useradd -m -u 1000 user
+
+USER user
+
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+COPY --chown=user . $HOME/app
+
+CMD ["streamlit", "run", "app.py"]

README.md

@@ -0,0 +1,12 @@
+---
+title: Container Test
+emoji: 📚
+colorFrom: gray
+colorTo: red
+sdk: docker
+pinned: false
+app_port: 8501
+duplicated_from: projects-jeru/container-test
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,31 @@
+from t5 import t5
+import torch
+import streamlit as st
+from streamlit import session_state as ses
+import timeit
+
+
+if "page" not in ses:
+    ses["page"] = "run"
+    ses.out = ""
+    ses.time = ""
+
+
+def __run_t5():
+    try:
+        ses.out = t5(ses.input_text)
+    except:
+        ses.out = "error"
+
+
+def __test_t5():
+    try:
+        ses.time = timeit.timeit(__run_t5(), globals=globals(), number=20)
+    except:
+        ses.out = "error"
+
+
+st.text_input("enter text", key="input_text", on_change=__run_t5)
+st.code(ses.out, language="json")
+if st.button("average time", on_click=__test_t5):
+    st.metric("time", ses.time)

requirements.txt

@@ -0,0 +1,6 @@
+streamlit==1.20.*
+requests==2.27.*
+sentencepiece==0.1.*
+torch==1.11.*
+transformers==4.*
+uvicorn[standard]==0.17.*

t5.py

@@ -0,0 +1,8 @@
+from transformers import pipeline
+
+pipe_flan = pipeline(model="bigscience/bloom-560m")
+
+
+def t5(input):
+    output = pipe_flan(input)
+    return output