DonImages commited on
Commit
dfee4d1
·
verified ·
1 Parent(s): a51aea0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 # Return seed to keep track of the used seed
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",
 
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",