update

Dockerfile

@@ -4,7 +4,7 @@ FROM python:3.11
 # Set the working directory to /code
 WORKDIR /code
 
-RUN apt-get update && apt-get install -y screen git
+RUN apt-get update && apt-get install -y screen git git-lfs
 RUN git lfs install
 
 # Copy the current directory contents into the container at /code

start.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
 
-screen -S script1 -dm bash -c 'python3 gradio_demo/seed_llama_flask.py --image_transform configs/transform/clip_transform.yaml --tokenizer configs/tokenizer/seed_llama_tokenizer.yaml --model configs/llm/seed_llama_14b.yaml   --port 7890 --llm_device cuda:0 --tokenizer_device cuda:0 --offload_tokenizer; exec bash'
-screen -S script2 -dm bash -c 'python3 gradio_demo/seed_llama_gradio.py; exec bash'
+# screen -S script1 -dm bash -c 'python3 gradio_demo/seed_llama_flask.py --image_transform configs/transform/clip_transform.yaml --tokenizer configs/tokenizer/seed_llama_tokenizer.yaml --model configs/llm/seed_llama_14b.yaml   --port 7890 --llm_device cuda:0 --tokenizer_device cuda:0 --offload_tokenizer; exec bash'
+screen -S script2 -dm bash -c 'python3 gradio_demo/seed_llama_gradio.py --server_name 0.0.0.0 --server_port 7860; exec bash'
 
 tail -f /dev/null