fix docker, links

Dockerfile

@@ -1,14 +1,12 @@
 FROM continuumio/miniconda3
 
-WORKDIR /code
+WORKDIR /
 
 RUN apt-get update && apt-get install --no-install-recommends -y \
   build-essential \
   libxml2 \
   && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-COPY ./requirements.txt /code/requirements.txt
-
 # Install dependencies
 RUN conda create -n pose python=3.8 
 
@@ -23,7 +21,6 @@ RUN conda install -c conda-forge cudatoolkit-dev -y
 ENV CUDA_HOME=$CONDA_PREFIX/pkgs/cuda-toolkit/
 RUN pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"
 
-
 RUN wget https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors -P pretrained_models
 RUN wget https://huggingface.co/stabilityai/sdxl-vae/resolve/main/sdxl_vae.safetensors -P pretrained_models
 

app.py

@@ -187,7 +187,7 @@ BASE_CONFIG = "configs/train_co3d_concept.yaml"
 BASE_CKPT = "pretrained-models/sd_xl_base_1.0.safetensors"
 
 start_time = time.time()
-base_model = load_base_model(BASE_CONFIG, ckpt=BASE_CKPT, verbose=False)
+# base_model = load_base_model(BASE_CONFIG, ckpt=BASE_CKPT, verbose=False)
 print(f"Time taken to load base model: {time.time() - start_time:.2f}s")
 
 global curr_camera_dict 
@@ -327,17 +327,24 @@ with gr.Blocks(head=head,
     gr.HTML("""
         <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
             <div>
-                <h1>Customizing Text-to-Image Diffusion with Camera Viewpoint Control</h1>
+                <h2><a href='https://customdiffusion360.github.io/index.html'>Customizing Text-to-Image Diffusion with Camera Viewpoint Control</a></h2>
             </div>
         </div>
-        <div>
-        </br>
+        <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
+            <a href='https://customdiffusion360.github.io/index.html' style="padding: 10px;">
+               <img src='https://img.shields.io/badge/Project%20Page-8A2BE2'>
+            </a>
+            <a class="link" href='https://github.com/customdiffusion360/custom-diffusion360' style="padding: 10px;">
+                <img src='https://img.shields.io/badge/Github-%23121011.svg'>
+            </a>
         </div>
         <hr></hr>
     """,
     visible=True
     )
 
+    gr
+
 
     if SPACE_ID == ORIGINAL_SPACE_ID:
         gr.Markdown(SHARED_UI_WARNING)