Update app.py

app.py

@@ -18,7 +18,7 @@ device = "cuda"
 cv2.setNumThreads(1)
 
 @spaces.GPU(enable_queue=True)
-def generate_image(images, prompt, negative_prompt,stylecode,strength,additional_prompt,  progress=gr.Progress(track_tqdm=True)):
+def generate_image(images, prompt, negative_prompt,stylecode,strength,additional_prompt, seed, progress=gr.Progress(track_tqdm=True)):
 
     
     full_prompt=prompt + additional_prompt
@@ -31,7 +31,7 @@ def generate_image(images, prompt, negative_prompt,stylecode,strength,additional
     # Calculate new width and height
    
 
-    image = use_stylecode(ip_ckpt,image_path="",prompt=prompt,negative_prompt=negative_prompt,num_inference_steps=20,image=image,stylecode=stylecode)
+    image = use_stylecode(ip_ckpt,image_path="",prompt=prompt,negative_prompt=negative_prompt,num_inference_steps=20,image=image,stylecode=stylecode,seed=seed)
     
     yield image
 
@@ -74,7 +74,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Accordion(open=False, label="Advanced Options"):
                 strength = gr.Slider(label="strength", value=0.9, step=0.05, minimum=0, maximum=1)
                 additional_prompt = gr.Textbox(label="additional prompt",placeholder ="highly detailed ,simple background")
-
+                seed = gr.Number(label="seed",placeholder = -1)
             
 
             submit = gr.Button("Use stylecode")
@@ -84,7 +84,7 @@ with gr.Blocks(css=css) as demo:
         stylecode_gen.click(fn=make_stylecode_gui,inputs=[files],outputs=stylecode)
         
         submit.click(fn=generate_image,
-                    inputs=[files, prompt, negative_prompt,stylecode,strength,additional_prompt],
+                    inputs=[files, prompt, negative_prompt,stylecode,strength,additional_prompt,seed],
                     outputs=gallery)
     
     gr.Markdown("This demo includes extra features to mitigate the implicit bias of the model and prevent explicit usage of it to generate content with faces of people, including third parties, that is not safe for all audiences, including naked or semi-naked people.")