Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-# 使用Ubuntu作为基础镜像
-FROM ubuntu:20.04
+# 使用PyTorch为基础镜像以支持Hugging Face
+FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-runtime
 
 # 避免交互式提示
 ENV DEBIAN_FRONTEND=noninteractive
@@ -9,15 +9,27 @@ 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
 
-# 安装必要的包
+# 安装系统依赖
 RUN apt-get update && apt-get install -y \
     wget \
     unzip \
-    openjdk-11-jdk \
     git \
     curl \
+    openjdk-11-jdk \
+    python3-pip \
     && rm -rf /var/lib/apt/lists/*
 
+# 安装Hugging Face transformers和相关依赖
+RUN pip3 install --no-cache-dir \
+    transformers \
+    datasets \
+    scikit-learn \
+    pandas \
+    numpy \
+    torch \
+    accelerate \
+    sentencepiece
+
 # 创建Android SDK目录
 RUN mkdir -p ${ANDROID_HOME}/cmdline-tools
 
@@ -39,20 +51,24 @@ RUN sdkmanager \
     "extras;google;m2repository"
 
 # 设置工作目录
-WORKDIR /app
+WORKDIR /workspace
 
-# 添加Gradle包装器
-COPY gradlew .
-COPY gradle gradle
-COPY build.gradle .
-COPY settings.gradle .
-COPY app app
+# 安装额外的Python包用于模型转换和优化
+RUN pip3 install --no-cache-dir \
+    onnx \
+    onnxruntime \
+    tensorflow-lite \
+    android-auto-bundle
 
-# 给予Gradle包装器执行权限
-RUN chmod +x ./gradlew
+# 配置Git（用于从Hugging Face下载模型）
+RUN git config --global credential.helper store
 
-# 预下载依赖
-RUN ./gradlew --no-daemon dependencies
+# 添加常用的启动脚本
+COPY start.sh /workspace/
+RUN chmod +x /workspace/start.sh
 
 # 设置默认命令
-CMD ["./gradlew", "assembleDebug"]
+CMD ["/bin/bash"]
+
+# 添加卷标用于持久化数据
+VOLUME ["/workspace/models", "/workspace/android-project"]