Update app.py

app.py

@@ -51,12 +51,9 @@ def infer(
     num_inference_steps=40,
     progress=gr.Progress(track_tqdm=True),
 ):
-
-    if randomize_seed:  
+    if randomize_seed:
         seed = random.randint(0, MAX_SEED)  # Generate a new seed if randomize_seed is True
-    else:
-        seed = seed  # Use the user-provided seed
-
+    
     generator = torch.Generator(device=device).manual_seed(seed)  # Ensure the generator is on the correct device
     
     try:
@@ -69,7 +66,7 @@ def infer(
             height=height,
             generator=generator,
         ).images[0]
-        return image, seed
+        return image, seed  # Return seed to keep track of the used seed
     except Exception as e:
         print(f"Error during image generation: {e}") # Print error for debugging
         return f"Error: {e}", seed # Return error to Gradio interface