Update app.py

app.py

@@ -62,7 +62,7 @@ def is_civitai_url(url):
 
 @spaces.GPU
 def generate_image(prompt, negative_prompt, lora_url, num_inference_steps=30, guidance_scale=7.0, 
-                  model="Real6.0", num_images=1, width=512, height=512):
+                  model="Real6.0", num_images=1, width=512, height=512,seed=None):
     
     if model == "Real5.0":
         model_id = "SG161222/Realistic_Vision_V5.0_noVAE"
@@ -147,6 +147,11 @@ def generate_image(prompt, negative_prompt, lora_url, num_inference_steps=30, gu
         use_karras_sigmas=True
     )
 
+    if seed is None:
+        seed = random.randint(0, 2**32 - 1)
+    
+    generator = torch.manual_seed(seed)
+
     text_inputs = tokenizer(
         prompt,
         padding="max_length",
@@ -175,10 +180,11 @@ def generate_image(prompt, negative_prompt, lora_url, num_inference_steps=30, gu
         guidance_scale=guidance_scale,
         width=width,
         height=height,
-        num_images_per_prompt=num_images
+        num_images_per_prompt=num_images,
+        generator=generator
     )
     
-    return result.images
+    return result.images,seed
 
 def clean_lora_cache():
     """Clean the LoRA cache directory"""
@@ -259,6 +265,7 @@ with gr.Blocks() as demo:
                     step=0.5,
                     label="Guidance Scale"
                 )
+                seed_input = gr.Number(value=random.randint(0, 2**32 - 1), label="Seed (optional)")
         
         with gr.Column():
             # Output component
@@ -269,13 +276,14 @@ with gr.Blocks() as demo:
                 columns=2,
                 rows=2
             )
+            seed_display = gr.Textbox(label="Seed Used", interactive=False)
     
     # Connect the interface to the generation function
     generate_button.click(
         fn=generate_image,
         inputs=[prompt, negative_prompt, lora_input, steps_slider, guidance_slider, 
-                model, num_images, width, height],
-        outputs=gallery
+                model, num_images, width, height,seed_input],
+        outputs=[gallery,seed]
     )
     
     # Connect clear cache button