Add Dockerfile

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.8.15
+
+RUN useradd -ms /bin/bash admin
+
+ENV WORKDIR=/code
+WORKDIR $WORKDIR
+RUN chown -R admin:admin $WORKDIR
+RUN chmod 755 $WORKDIR
+
+COPY requirements.txt $WORKDIR/requirements.txt
+
+RUN pip install gradio --no-cache-dir
+RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
+
+COPY . .
+
+RUN sh deform_setup.sh
+
+USER admin
+
+EXPOSE 7860
+
+ENTRYPOINT ["python", "gradio_app.py"]

README.md

@@ -3,12 +3,13 @@ title: OneFormer
 emoji: 🎗️
 colorFrom: red
 colorTo: blue
-sdk: gradio
-sdk_version: 3.1.4
-app_file: gradio_app.py
+sdk: docker
+app_port: 7860
+# sdk_version: docker
+# app_file: gradio_app.py
 pinned: false
 license: mit
-python_version: 3.8.15
+# python_version: 3.8.15
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

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!")