Update app.py

app.py

@@ -51,10 +51,14 @@ def infer(
     num_inference_steps=40,
     progress=gr.Progress(track_tqdm=True),
 ):
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator().manual_seed(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:
         image = pipe(
             prompt=prompt,
@@ -70,10 +74,6 @@ def infer(
         print(f"Error during image generation: {e}") # Print error for debugging
         return f"Error: {e}", seed # Return error to Gradio interface
 
-# ... (rest of your Gradio code - examples, CSS, etc. - same as before)
-examples = [
-        "A capybara wearing a suit holding a sign that reads Hello World",
-]
 
 css = """
 #col-container {