updates

Dockerfile

@@ -14,6 +14,10 @@ COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+RUN pip install diffusers transformers accelerate scipy safetensors
+
+
+
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 
@@ -30,12 +34,14 @@ WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
 # Download and uzip truepic-size from Google Drive
-RUN gdown 1MdXMu8xSuG8WClY3JSgtL3Vy1IWTmWZO
-RUN tar -xf truepic-sign-v0.1.0-ubuntu-latest.tar.gz
+RUN pip install gdown
+RUN --mount=type=secret,id=truepic_sign_gdoc_id,mode=0444,required=true \
+   gdown --id $(cat /run/secrets/truepic_sign_gdoc_id)
+RUN tar -xf truepic-sign-v0.2.0-ubuntu-20.04.tar.gz
 RUN chmod +x truepic-sign
 
 WORKDIR $HOME/app
 
 # COPY . .
 
-CMD ["python3", "app.py"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -1,8 +1,8 @@
 fastapi==0.74.*
 requests==2.27.*
+sentencepiece==0.1.*
 torch==1.11.*
 transformers==4.*
-gradio
-diffusers
-accelerate
-gdown
+uvicorn[standard]==0.17.*
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch