Update app.py
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ def infer(
|
|
53 |
):
|
54 |
if randomize_seed:
|
55 |
seed = random.randint(0, MAX_SEED) # Generate a new seed if randomize_seed is True
|
56 |
-
|
57 |
generator = torch.Generator(device=device).manual_seed(seed) # Ensure the generator is on the correct device
|
58 |
|
59 |
try:
|
@@ -66,10 +66,10 @@ def infer(
|
|
66 |
height=height,
|
67 |
generator=generator,
|
68 |
).images[0]
|
69 |
-
return image, seed #
|
70 |
except Exception as e:
|
71 |
-
print(f"Error during image generation: {e}")
|
72 |
-
return f"Error: {e}", seed
|
73 |
|
74 |
examples = [
|
75 |
"A capybara wearing a suit holding a sign that reads Hello World",
|
|
|
53 |
):
|
54 |
if randomize_seed:
|
55 |
seed = random.randint(0, MAX_SEED) # Generate a new seed if randomize_seed is True
|
56 |
+
|
57 |
generator = torch.Generator(device=device).manual_seed(seed) # Ensure the generator is on the correct device
|
58 |
|
59 |
try:
|
|
|
66 |
height=height,
|
67 |
generator=generator,
|
68 |
).images[0]
|
69 |
+
return image, seed # Don't return seed back to the UI
|
70 |
except Exception as e:
|
71 |
+
print(f"Error during image generation: {e}") # Print error for debugging
|
72 |
+
return f"Error: {e}", seed # Return error to Gradio interface
|
73 |
|
74 |
examples = [
|
75 |
"A capybara wearing a suit holding a sign that reads Hello World",
|