docker: change gradio env var

Dockerfile

@@ -21,8 +21,6 @@ RUN conda create -n pose python=3.8
 # Make RUN commands use the new environment:
 SHELL ["conda", "run", "-n", "pose", "/bin/bash", "-c"]
 
-# Set gradio env variables?
-
 RUN pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
 
 RUN pip install -r requirements.txt
@@ -34,4 +32,6 @@ 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
 
+ENV GRADIO_SERVER_NAME=0.0.0.0
+
 ENTRYPOINT ["conda", "run", "--no-capture-output", "-n", "pose", "python", "app.py"] 

README.md

@@ -7,6 +7,7 @@ sdk: docker
 app_port: 7860
 app_file: app.py
 pinned: false
+suggested_hardeware: a10g-large
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -153,7 +153,9 @@ MODELS_DIR = "pretrained-models/"
 
 def select_and_load_model(category, category_single_id):
     global current_data, current_model, base_model
-    current_model = None
+    del current_model
+    del current_data    
+    torch.cuda.empty_cache()
     current_model = copy.deepcopy(base_model)
 
     ### choose model checkpoint and config
@@ -167,7 +169,7 @@ def select_and_load_model(category, category_single_id):
     current_model, current_data = load_and_return_model_and_data(config, current_model,
                                                                     delta_ckpt=delta_ckpt
                                                                     )
-        
+    
     print(f"Time taken to load delta model: {time.time() - start_time:.2f}s")
     
     print("!!! model loaded")