Upload 6 files

Dockerfile

@@ -4,17 +4,10 @@ RUN apt update -y
 RUN apt install -y wget zip unzip
 
 COPY . .
+RUN chmod 777 text2sign.sh
 RUN pip3 install -r requirements.txt
 
-
-# Spoken 2 Sign
-RUN pip3 install --pre -f https://dist.mxnet.io/python 'mxnet==2.0.0b20220206'
-
-RUN wget https://github.com/SignBridgeApp/models/releases/download/v0.0/spoken2symbol.zip
-RUN unzip spoken2symbol.zip
-RUN rm spoken2symbol.zip
-RUN apt install -y sentencepiece
-
+RUN ./text2sign.sh
 
 # Start
 EXPOSE 7860

app.py

@@ -15,7 +15,7 @@ def translate_text():
 
 
 @route("/sign2img")
-def covert_text():
+def convert_text():
     sign = request.query.get("sign", None)
     if not sign:
         response.status_code = 400
@@ -28,7 +28,7 @@ def covert_text():
 
 @route("/")
 def index():
-    return {"Server": "OK"}
+    return {"signbridge-server": "OK"}
 
 
 if __name__ == "__main__":

text2sign.py

@@ -7,7 +7,7 @@ warnings.filterwarnings("ignore")
 
 
 DEVICE = mx.cpu()
-MODEL_FOLDER = "spoken2symbol"
+MODEL_FOLDER = "text2sign"
 SPM_PATH = MODEL_FOLDER + "/spm.model"
 
 

text2sign.sh

@@ -0,0 +1,23 @@
+#!/bin/bash
+
+URL="https://github.com/SignBridgeApp/models/releases/download/v0.0/text2sign.zip"
+ZIP_FILE="text2sign.zip"
+TARGET_DIR="text2sign"
+
+if [ ! -d "$TARGET_DIR" ]; then
+    echo "Downloading and extracting $ZIP_FILE..."
+
+	if [ ! -d "$ZIP_FILE" ]; then
+    	wget -q "$URL"
+	else
+    	echo "Zip $ZIP_FILE already exists. Skipping download."
+	fi
+
+	unzip -q "$ZIP_FILE" && rm "$ZIP_FILE"
+fi
+else
+    echo "Directory $TARGET_DIR already exists. Skipping download and extraction."
+fi
+
+pip3 install --pre -f https://dist.mxnet.io/python 'mxnet==2.0.0b20220206'
+apt install -y sentencepiece