Update app.py

app.py

@@ -79,7 +79,7 @@ pipe.FacialEncoder.to(device)
     
 @spaces.GPU
 def process(selected_template_images,costum_image,prompt
-        ,negative_prompt,prompt_selected,retouching,model_selected_tab,prompt_selected_tab,width,height,merge_steps):
+        ,negative_prompt,prompt_selected,retouching,model_selected_tab,prompt_selected_tab,width,height,merge_steps,seed_set):
     
     if model_selected_tab==0:
         select_images = load_image(Image.open(selected_template_images))
@@ -95,6 +95,7 @@ def process(selected_template_images,costum_image,prompt
 
     # hyper-parameter
     num_steps = 50
+    seed_set = torch.randint(0, 1000, (1,)).item()
     # merge_steps = 30
             
     @torch.inference_mode()
@@ -136,8 +137,8 @@ def process(selected_template_images,costum_image,prompt
     negtive_prompt_group="((cross-eye)),((cross-eyed)),(((NFSW))),(nipple),((((ugly)))), (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))). out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck)))"
     negative_prompt = negative_prompt + negtive_prompt_group
     
-    seed = torch.randint(0, 1000, (1,)).item()
-    generator = torch.Generator(device=device).manual_seed(seed)
+    # seed = torch.randint(0, 1000, (1,)).item()
+    generator = torch.Generator(device=device).manual_seed(seed_set)
 
     images = pipe(
         prompt=prompt,
@@ -214,6 +215,7 @@ with gr.Blocks(title="ConsistentID Demo") as demo:
             width.release(lambda x,y: min(1280-x,y), inputs=[width,height], outputs=[height])
             height.release(lambda x,y: min(1280-y,x), inputs=[width,height], outputs=[width])
             merge_steps = gr.Slider(label="step starting to merge facial details(30 is recommended)",minimum=10,maximum=50,value=30,step=1)
+            seed_set = gr.Slider(label="set the random seed for different results",minimum=1,maximum=2147483647,value=2024,step=1)
             
             btn = gr.Button("Run")
         with gr.Column():
@@ -225,6 +227,6 @@ with gr.Blocks(title="ConsistentID Demo") as demo:
                 - Due to insufficient graphics memory on the demo server, there is an upper limit on the resolution for generating samples. We will support the generation of SDXL as soon as possible<br/><br/>
                 ''')
         btn.click(fn=process, inputs=[selected_template_images,costum_image,prompt,nagetive_prompt,prompt_selected,retouching
-            ,model_selected_tab,prompt_selected_tab,width,height,merge_steps], outputs=out)
+            ,model_selected_tab,prompt_selected_tab,width,height,merge_steps,seed_set], outputs=out)
 
 demo.launch()