Fix user

Dockerfile

@@ -45,6 +45,9 @@ RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd o
 
 RUN sh deform_setup.sh
 
+USER user
+RUN sh deform_setup.sh
+
 USER user
 
 EXPOSE 7860

deform_setup.sh

@@ -1,9 +1,9 @@
 #!/usr/bin/env bash
-# nvidia-smi
+
 # ln -s ./oneformer/modeling/pixel_decoder/ops/ ./
 # ls
 # cd ops/ && bash make.sh && cd ..
-
+nvidia-smi
 echo '----------------------------------------------------------------'
 echo '----------------------------------------------------------------'
 pip3 freeze | grep MultiScaleDeformableAttention

gradio_app.py

@@ -1,7 +1,7 @@
 import subprocess
 import torch
-# if torch.cuda.is_available():
-#    subprocess.run('sh deform_setup.sh', shell=True)
+if torch.cuda.is_available():
+   subprocess.run('sh deform_setup.sh', shell=True)
 
 print("Installed the dependencies!")