Update files.

app.py

@@ -7,8 +7,6 @@ from webui import (
 )
 
 
-import os
-os.environ["no_proxy"] = "localhost,127.0.0.1,::1"
 
 
 def main():
@@ -31,8 +29,9 @@ def main():
 
             with gr.TabItem("Texture Synthesis", id='tab_texture_syn'):
                 create_interface_texture_synthesis(runner=runner)
-            
-        demo.launch(share=False, debug=False)
+
+        demo.queue().lanuch()
+        # demo.launch(share=False, debug=False)
 
 
 if __name__ == '__main__':

webui/runner.py

@@ -9,6 +9,7 @@ from pipeline_sdxl import ADPipeline as ADXLPipeline
 from utils import Controller
 
 import os
+import spaces
 
 
 class Runner:
@@ -37,6 +38,7 @@ class Runner:
         image = image.resize(size, Image.BICUBIC)
         return to_tensor(image).unsqueeze(0)
 
+    @spaces.GPU
     def run_style_transfer(self, content_image, style_image, seed, num_steps, lr, content_weight, mixed_precision, model, **kwargs):
         self.load_pipeline(model)
 
@@ -65,6 +67,7 @@ class Runner:
         torch.cuda.empty_cache()
         return [output_image]
 
+    @spaces.GPU
     def run_style_t2i_generation(self, style_image, prompt, negative_prompt, guidance_scale, height, width, seed, num_steps, iterations, lr, num_images_per_prompt, mixed_precision, is_adain, model):
         self.load_pipeline(model)
 
@@ -100,6 +103,7 @@ class Runner:
         torch.cuda.empty_cache()
         return output_images
 
+    @spaces.GPU
     def run_texture_synthesis(self, texture_image, height, width, seed, num_steps, iterations, lr, mixed_precision, num_images_per_prompt, synthesis_way,model):
         self.load_pipeline(model) 
 

webui/tab_style_t2i.py

@@ -18,13 +18,13 @@ def create_interface_style_t2i(runner):
                 prompt = gr.Textbox(label='Prompt', value='A rocket')
                 negative_prompt = gr.Textbox(label='Negative Prompt', value='')
 
-                base_model_list = ['stable-diffusion-v1-5', 'stable-diffusion-xl-base-1.0']
-                model = gr.Radio(choices=base_model_list, label='Select a Base Model', value='stable-diffusion-xl-base-1.0')
+                base_model_list = ['stable-diffusion-v1-5/stable-diffusion-v1-5', 'stabilityai/stable-diffusion-xl-base-1.0']
+                model = gr.Radio(choices=base_model_list, label='Select a Base Model', value='stabilityai/stable-diffusion-xl-base-1.0')
 
                 run_button = gr.Button(value='Run')
                 
                 gr.Examples(
-                    [[Image.open('./webui/images/image_02_01.jpg').convert('RGB'), 'A rocket', 'stable-diffusion-xl-base-1.0']],
+                    [[Image.open('./webui/images/image_02_01.jpg').convert('RGB'), 'A rocket', 'stabilityai/stable-diffusion-xl-base-1.0']],
                     [style_image, prompt, model]
                 )
 

webui/tab_style_transfer.py

@@ -27,8 +27,8 @@ def create_interface_style_transfer(runner):
                     content_weight = gr.Slider(label='Content Weight', minimum=0., maximum=1., value=0.25, step=0.001)
                     mixed_precision = gr.Radio(choices=['bf16', 'no'], value='bf16', label='Mixed Precision')
                     
-                    base_model_list = ['stable-diffusion-v1-5',]
-                    model = gr.Radio(choices=base_model_list, label='Select a Base Model', value='stable-diffusion-v1-5')
+                    base_model_list = ['stable-diffusion-v1-5/stable-diffusion-v1-5',]
+                    model = gr.Radio(choices=base_model_list, label='Select a Base Model', value='stable-diffusion-v1-5/stable-diffusion-v1-5')
 
             with gr.Column():
                 gr.Markdown('#### Output Image:\n')

webui/tab_texture_synthesis.py

@@ -28,8 +28,8 @@ def create_interface_texture_synthesis(runner):
                     mixed_precision = gr.Radio(choices=['bf16', 'no'], value='bf16', label='Mixed Precision')
                     num_images_per_prompt = gr.Slider(label='Num Images Per Prompt', minimum=1, maximum=10, value=1, step=1)
                     
-                    base_model_list = ['stable-diffusion-v1-5',]
-                    model = gr.Radio(choices=base_model_list, label='Select a Base Model', value='stable-diffusion-v1-5')
+                    base_model_list = ['stable-diffusion-v1-5/stable-diffusion-v1-5',]
+                    model = gr.Radio(choices=base_model_list, label='Select a Base Model', value='stable-diffusion-v1-5/stable-diffusion-v1-5')
                     synthesis_way = gr.Radio(['Sampling', 'MultiDiffusion'], label='Synthesis Way', value='MultiDiffusion')
 
             with gr.Column():