Update app.py

app.py

@@ -35,10 +35,20 @@ def load_lora_model(pipe, lora_model_path):
 pipe = load_lora_model(pipe, lora_model_path)
 
 # Function to generate an image from a text prompt
-def generate_image(prompt):
-    image = pipe(prompt, height=1080, width=1080).images[0]
+def generate_image(prompt, seed=None):
+    generator = torch.manual_seed(seed) if seed is not None else None
+    image = pipe(prompt, height=1080, width=1080, generator=generator).images[0]
     return image
 
+
 # Gradio interface
-iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.Textbox(label="Enter your prompt"),  # For the prompt
+        gr.Number(label="Enter a seed (optional)", value=None),  # For the seed
+    ],
+    outputs="image"
+)
 iface.launch()
+