Update Dockerfile

Dockerfile

@@ -8,6 +8,7 @@ ENV DEBIAN_FRONTEND=noninteractive
 ENV ANDROID_HOME=/opt/android-sdk
 ENV ANDROID_SDK_ROOT=$ANDROID_HOME
 ENV PATH=$PATH:$ANDROID_HOME/platform-tools:$ANDROID_HOME/cmdline-tools/latest/bin
+ENV PYTHONUNBUFFERED=1
 
 # 安装系统依赖
 RUN apt-get update && apt-get install -y \
@@ -17,6 +18,7 @@ RUN apt-get update && apt-get install -y \
     curl \
     openjdk-11-jdk \
     python3-pip \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
 
 # 安装Hugging Face transformers和相关依赖
@@ -28,7 +30,8 @@ RUN pip3 install --no-cache-dir \
     numpy \
     torch \
     accelerate \
-    sentencepiece
+    sentencepiece \
+    tensorboard
 
 # 创建Android SDK目录
 RUN mkdir -p ${ANDROID_HOME}/cmdline-tools
@@ -57,18 +60,20 @@ WORKDIR /workspace
 RUN pip3 install --no-cache-dir \
     onnx \
     onnxruntime \
-    tensorflow-lite \
+    tensorflow \
+    tensorflow-hub \
     android-auto-bundle
 
-# 配置Git（用于从Hugging Face下载模型）
+# 配置Git
 RUN git config --global credential.helper store
 
-# 添加常用的启动脚本
+# 创建必要的目录
+RUN mkdir -p /workspace/models /workspace/android-project
+
+# 复制启动脚本
 COPY start.sh /workspace/
 RUN chmod +x /workspace/start.sh
 
-# 设置默认命令
-CMD ["/bin/bash"]
-
-# 添加卷标用于持久化数据
-VOLUME ["/workspace/models", "/workspace/android-project"]
+# 设置启动命令
+ENTRYPOINT ["/workspace/start.sh"]
+CMD ["/bin/bash"]