update

.vscode/launch.json

@@ -0,0 +1,27 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python Debugger: Current File",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${file}",
+            "console": "integratedTerminal"
+        },
+        {
+            "name": "Python Debugger: app.py",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "app.py",
+            "console": "integratedTerminal",
+            "justMyCode": false,
+            "env": {
+                "CUDA_VISIBLE_DEVICES": "7",
+                "GRADIO_SERVER_PORT": "7864",
+            }
+        }
+    ]
+}

DockerFile

@@ -1,23 +1,24 @@
-# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
-# you will also find guides on how best to write your Dockerfile
-
-FROM python:3.11
+FROM continuumio/anaconda3:main
 
 WORKDIR /code
+COPY ./environment.yml /code/environment.yml
 
-COPY ./pre-requirements.txt /code/pre-requirements.txt
-
-RUN pip install --no-cache-dir --upgrade -r /code/pre-requirements.txt
+# Create the environment using the environment.yml file
+RUN conda env create -f /code/environment.yml
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
-
 # Switch to the "user" user
 USER user
-
 # Set home to the user's home directory
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
+    PYTHONPATH=$HOME/app \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
@@ -25,4 +26,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["python", "app.py"]
+CMD ["./run.sh"]

install.sh

@@ -1,6 +1,7 @@
-conda activate base
-conda remove -n matgen-plus --all
-
+#!/bin/bash
+eval "$(conda shell.bash hook)"
+# conda activate base
+# conda remove -n matgen-plus --all
 
 conda create -n matgen-plus python=3.11
 conda activate matgen-plus
@@ -14,4 +15,4 @@ conda install pytorch3d -c pytorch -c conda-forge
 conda install -c conda-forge open-clip-torch pytorch-lightning
 pip install trimesh xatlas scikit-learn opencv-python omegaconf
 
-bash run.sh
+python app.py