Update Dockerfile

Dockerfile

@@ -20,7 +20,7 @@ WORKDIR /app
 
 # 安装依赖
 RUN pip install --upgrade pip
-RUN pip install torch transformers peft datasets accelerate bitsandbytes huggingface_hub
+RUN pip install torch transformers peft datasets accelerate bitsandbytes huggingface_hub gradio
 
 # 将代码和数据复制到容器中
 COPY . /app