update gradio

app.py

@@ -70,7 +70,7 @@ class GlobalText:
         self.personal_model_loaded = None
         self.lora_model_state_dict = {}
         self.device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-        self.refresh_stable_diffusion()
+        # self.refresh_stable_diffusion()
         self.refresh_personalized_model()
 
         self.reset_start_code()
@@ -393,14 +393,14 @@ def ui():
             with gr.Tab("Configs"):
 
                 with gr.Row():
-                    source_image = gr.Image(label="Source Image",  elem_id="img2maskimg", source="upload", interactive=True, type="pil", tool="sketch", image_mode="RGB", height=512)
-                    style_image = gr.Image(label="Style Image", elem_id="img2maskimg", source="upload", interactive=True, type="pil", tool="sketch", image_mode="RGB", height=512)
+                    source_image = gr.Image(label="Source Image",  elem_id="img2maskimg", sources="upload", interactive=True, type="pil",image_mode="RGB", height=512)
+                    style_image = gr.Image(label="Style Image", elem_id="img2maskimg", sources="upload", interactive=True, type="pil", image_mode="RGB", height=512)
                 with gr.Row():
                     prompt_textbox = gr.Textbox(label="Prompt", value='head', lines=1)
                     negative_prompt_textbox = gr.Textbox(label="Negative prompt", lines=1)
                     # output_dir = gr.Textbox(label="output_dir", value='./results/')
 
-                with gr.Row().style(equal_height=False):
+                with gr.Row(equal_height=False):
                     with gr.Column():
                         width_slider     = gr.Slider(label="Width", value=512, minimum=256, maximum=1024, step=64)
                         height_slider    = gr.Slider(label="Height", value=512, minimum=256, maximum=1024, step=64)
@@ -487,23 +487,22 @@ def ui():
                         sam_style_btn = gr.Button(value="SAM Style")
                         send_style_btn = gr.Button(value="Send Style")
                     with gr.Row():
-                        source_image_sam = gr.Image(label="Source Image SAM",  elem_id="SourceimgSAM", source="upload", interactive=True, type="pil", image_mode="RGB", height=512)
-                        style_image_sam = gr.Image(label="Style Image SAM", elem_id="StyleimgSAM", source="upload", interactive=True, type="pil", image_mode="RGB", height=512)
+                        source_image_sam = gr.Image(label="Source Image SAM",  elem_id="SourceimgSAM", sources="upload", interactive=True, type="pil", image_mode="RGB", height=512)
+                        style_image_sam = gr.Image(label="Style Image SAM", elem_id="StyleimgSAM", sources="upload", interactive=True, type="pil", image_mode="RGB", height=512)
 
                     with gr.Row():
                         source_image_with_points = gr.Image(label="source Image with points", elem_id="style_image_with_points", type="pil", image_mode="RGB", height=256)
-                        source_mask = gr.Image(label="Source Mask",  elem_id="img2maskimg", source="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
+                        source_mask = gr.Image(label="Source Mask",  elem_id="img2maskimg", sources="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
 
                         style_image_with_points = gr.Image(label="Style Image with points", elem_id="style_image_with_points", type="pil", image_mode="RGB", height=256)
-                        style_mask = gr.Image(label="Style Mask", elem_id="img2maskimg", source="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
-        
+                        style_mask = gr.Image(label="Style Mask", elem_id="img2maskimg", sources="upload", interactive=True, type="numpy", image_mode="RGB", height=256)
         gr.Examples(
-            [[os.path.join(os.path.dirname(__file__), "images/content/1.jpg"),
-              os.path.join(os.path.dirname(__file__), "images/style/1.jpg")],
-            
-            ],
-            [source_image, style_image]
+                [[os.path.join(os.path.dirname(__file__), "images/content/1.jpg"),
+                os.path.join(os.path.dirname(__file__), "images/style/1.jpg")]],
+                #/home/jin.liu/liujin/mycode/PortraitDiffusion/images/style/1.jpg
+                [source_image, style_image]
         )
+        
         inputs = [
             source_image, style_image, source_mask, style_mask,
             start_step, start_layer, Style_attn_step, 

app.sh

@@ -4,4 +4,4 @@ export CUDA_VISIBLE_DEVICES=$1
 
 echo "CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES"
 # export CUDA_VISIBLE_DEVICES=5
-python gapp.py 
+python app.py 

requirements.txt

@@ -7,4 +7,4 @@ pytorch_lightning
 torchvision
 pillow
 git+https://github.com/facebookresearch/segment-anything.git
-gradio==3.47
+gradio