fix dependency

README.md

@@ -3,9 +3,7 @@ title: DA-Transformer
 emoji: 😻
 colorFrom: red
 colorTo: gray
-sdk: gradio
-sdk_version: 3.28.3
-app_file: app.py
+sdk: docker
 pinned: false
 license: apache-2.0
 ---

dockerfile

@@ -0,0 +1,24 @@
+# Use the official Python 3.9 image
+FROM python:3.9
+
+RUN --mount=target=/root/packages.txt,source=packages.txt 	sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/sources.list && sed -i 's http://archive.ubuntu.com http://us-east-1.ec2.archive.ubuntu.com g' /etc/apt/sources.list && sed -i '/security/d' /etc/apt/sources.list && apt-get update &&     xargs -r -a /root/packages.txt apt-get install -y     && rm -rf /var/lib/apt/lists/*
+
+RUN pip install --no-cache-dir pip==22.3.1 &&     pip install --no-cache-dir         datasets         "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1"
+
+WORKDIR /home/user/app
+
+RUN sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/sources.list && sed -i 's http://archive.ubuntu.com http://us-east-1.ec2.archive.ubuntu.com g' /etc/apt/sources.list && sed -i '/security/d' /etc/apt/sources.list && apt-get update && apt-get install -y 	git 	git-lfs 	ffmpeg 	libsm6 	libxext6 	cmake 	libgl1-mesa-glx 	&& rm -rf /var/lib/apt/lists/* 	&& git lfs install
+
+RUN pip install --no-cache-dir Cython "gradio==3.28.3" "torch==1.10.1"
+
+RUN git clone --recurse-submodules https://github.com/thu-coai/DA-Transformer.git && git checkout demo && pip install -e .
+
+RUN cd dag_search && python3 setup.py build_ext --inplace && pip install -e .
+
+USER user
+
+COPY --link --chown=1000 --from=lfs /app /home/user/app
+
+COPY --link --chown=1000 ./ /home/user/app
+
+CMD ["python3", "app.py"]

pre-requirements.txt

@@ -1 +0,0 @@
-Cython

requirements.txt

@@ -1,3 +0,0 @@
-torch == 1.10.1
-git+https://github.com/thu-coai/DA-Transformer.git#egg=fairseq[demo]
-git+https://github.com/thu-coai/DAG-Search.git