praeclarumjj3 commited on
Commit
7476063
1 Parent(s): 3701f72

Fix image urls

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -6
  2. gradio_app.py +3 -3
Dockerfile CHANGED
@@ -28,7 +28,6 @@ WORKDIR $WORKDIR
28
  RUN chown -R user:user $WORKDIR
29
  RUN chmod -R 777 $WORKDIR
30
 
31
-
32
  COPY requirements.txt $WORKDIR/requirements.txt
33
  RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
34
 
@@ -39,8 +38,6 @@ ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
39
  RUN pip install ninja
40
 
41
  USER root
42
- # RUN chown -R user:user /usr
43
- # RUN chmod -R 777 /usr
44
  RUN chown -R user:user $HOME
45
  RUN chmod -R 777 $HOME
46
  RUN chown -R user:user $WORKDIR
@@ -54,9 +51,9 @@ USER user
54
  RUN sh deform_setup.sh
55
 
56
  RUN mkdir -p examples
57
- RUN wget https://github.com/SHI-Labs/OneFormer-Colab/blob/master/samples/ade20k.jpeg -P $WORKDIR/examples/
58
- RUN wget https://github.com/SHI-Labs/OneFormer-Colab/blob/master/samples/cityscapes.png -P $WORKDIR/examples/
59
- RUN wget https://github.com/SHI-Labs/OneFormer-Colab/blob/master/samples/coco.jpeg -P $WORKDIR/examples/
60
 
61
  USER user
62
 
 
28
  RUN chown -R user:user $WORKDIR
29
  RUN chmod -R 777 $WORKDIR
30
 
 
31
  COPY requirements.txt $WORKDIR/requirements.txt
32
  RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
33
 
 
38
  RUN pip install ninja
39
 
40
  USER root
 
 
41
  RUN chown -R user:user $HOME
42
  RUN chmod -R 777 $HOME
43
  RUN chown -R user:user $WORKDIR
 
51
  RUN sh deform_setup.sh
52
 
53
  RUN mkdir -p examples
54
+ RUN wget https://praeclarumjj3.github.io/files/ade20k.jpeg -P $WORKDIR/examples/
55
+ RUN wget https://praeclarumjj3.github.io/files/cityscapes.png -P $WORKDIR/examples/
56
+ RUN wget https://praeclarumjj3.github.io/files/coco.jpeg -P $WORKDIR/examples/
57
 
58
  USER user
59
 
gradio_app.py CHANGED
@@ -189,10 +189,10 @@ title = "OneFormer: One Transformer to Rule Universal Image Segmentation"
189
 
190
  description = "<p style='color: #E0B941; font-size: 16px; font-weight: w600; text-align: center'> <a style='color: #E0B941;' href='https://praeclarumjj3.github.io/oneformer/' target='_blank'>Project Page</a> | <a style='color: #E0B941;' href='https://arxiv.org/abs/2211.06220' target='_blank'>OneFormer: One Transformer to Rule Universal Image Segmentation</a> | <a style='color: #E0B941;' href='https://github.com/SHI-Labs/OneFormer' target='_blank'>Github</a></p>" \
191
  + "<p style='color:royalblue; margin: 10px; font-size: 16px; font-weight: w400;'> \
192
- [Note: Inference on CPU may take upto 2 minutes.] This is the official gradio demo for our paper <span style='color:#E0B941;'>OneFormer: One Transformer to Rule Universal Image Segmentation</span> To use OneFormer: <br> \
193
  (1) <span style='color:#E0B941;'>Upload an Image</span> or <span style='color:#E0B941;'> select a sample image from the examples</span> <br> \
194
- (2) Select the value of the <span style='color:#E0B941;'>Task Token Input</span> <br>\
195
- (3) Select the <span style='color:#E0B941;'>Model</span> </p>"
196
 
197
  # article =
198
 
 
189
 
190
  description = "<p style='color: #E0B941; font-size: 16px; font-weight: w600; text-align: center'> <a style='color: #E0B941;' href='https://praeclarumjj3.github.io/oneformer/' target='_blank'>Project Page</a> | <a style='color: #E0B941;' href='https://arxiv.org/abs/2211.06220' target='_blank'>OneFormer: One Transformer to Rule Universal Image Segmentation</a> | <a style='color: #E0B941;' href='https://github.com/SHI-Labs/OneFormer' target='_blank'>Github</a></p>" \
191
  + "<p style='color:royalblue; margin: 10px; font-size: 16px; font-weight: w400;'> \
192
+ [Note: Inference on CPU may take upto 2 minutes.] This is the official gradio demo for our paper <span style='color:#E0B941;'>OneFormer: One Transformer to Rule Universal Image Segmentation</span> To use <span style='color:#E0B941;'>OneFormer</span>: <br> \
193
  (1) <span style='color:#E0B941;'>Upload an Image</span> or <span style='color:#E0B941;'> select a sample image from the examples</span> <br> \
194
+ (2) Select the value of the <span style='color:#E0B941;'>Task Token Input</span>. <br>\
195
+ (3) Select the <span style='color:#E0B941;'>Model</span> and <span style='color:#E0B941;'>Backbone</span>. </p>"
196
 
197
  # article =
198