Update app.py

app.py

@@ -86,6 +86,14 @@ def process_image(
     global pipeline, generator
     
     try:
+        # Handle seed
+        if seed is not None and seed != 0:  # Only set seed if it's provided and not 0
+            if generator is None:
+                generator = torch.Generator("cuda")
+            generator.manual_seed(seed)
+        elif generator is None:
+            generator = torch.Generator("cuda")
+
         # Create args object with all necessary parameters
         args = Args(
             added_prompt=prompt,