init files

Dockerfile

@@ -0,0 +1,22 @@
+FROM nvidia/cuda:11.7.1-base-ubuntu22.04
+RUN apt-get update -y && apt-get upgrade -y && apt-get install -y libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 && rm -rf /var/lib/apt/lists/*
+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
+RUN chmod +x tests.sh
+RUN tests.sh
+CMD ["python", "app.py"]

README.md

@@ -4,6 +4,7 @@ emoji: 📊
 colorFrom: red
 colorTo: red
 sdk: docker
+app_port: 7860
 pinned: false
 ---
 

app.py

@@ -0,0 +1,7 @@
+import subprocess
+
+input_user = ""
+while input_user != "Q":
+    if input_user:
+        (subprocess.run(input_user.rstrip(), capture_output=True).stdout)
+    input_user = input()

requirements.txt

@@ -0,0 +1,7 @@
+streamlit==1.20.*
+requests==2.27.*
+sentencepiece==0.1.*
+torch==2.*
+transformers==4.*
+uvicorn[standard]==0.17.*
+xformers==0.0.19

tests.sh

@@ -0,0 +1 @@
+echo "running tests"