Compile Worked, fix path data

Dockerfile

@@ -31,32 +31,31 @@ RUN chmod -R 777 $WORKDIR
 
 COPY requirements.txt $WORKDIR/requirements.txt
 COPY . .
-RUN cd $WORKDIR/oneformer/data && ls
 
-# RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
 
-# ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
+ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
 
 
-# RUN pip install ninja
+RUN pip install ninja
 
-# USER root
-# RUN chown -R user:user /usr
-# RUN chmod -R 777 /usr
-# RUN chown -R user:user $HOME
-# RUN chmod -R 777 $HOME
-# RUN chown -R user:user $WORKDIR
-# RUN chmod -R 777 $WORKDIR
+USER root
+RUN chown -R user:user /usr
+RUN chmod -R 777 /usr
+RUN chown -R user:user $HOME
+RUN chmod -R 777 $HOME
+RUN chown -R user:user $WORKDIR
+RUN chmod -R 777 $WORKDIR
 
-# USER user
-# RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ install --user && cd ..
-# RUN sh deform_setup.sh
+USER user
+RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ install --user && cd ..
+RUN sh deform_setup.sh
 
-# USER user
-# RUN sh deform_setup.sh
+USER user
+RUN sh deform_setup.sh
 
-# USER user
+USER user
 
-# EXPOSE 7860
+EXPOSE 7860
 
-# ENTRYPOINT ["python", "gradio_app.py"]
+ENTRYPOINT ["python", "gradio_app.py"]

oneformer/data/tokenizer.py

@@ -37,8 +37,8 @@ import torch
 
 @lru_cache()
 def default_bpe():
-    url = 'https://github.com/SHI-Labs/OneFormer/blob/main/oneformer/data/bpe_simple_vocab_16e6.txt.gz'
-    wget.download(url, out=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bpe_simple_vocab_16e6.txt.gz'))
+    # url = 'https://github.com/SHI-Labs/OneFormer/blob/main/oneformer/data/bpe_simple_vocab_16e6.txt.gz'
+    # wget.download(url, out=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bpe_simple_vocab_16e6.txt.gz'))
     return os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bpe_simple_vocab_16e6.txt.gz')
 
 @lru_cache()