cavargas10 commited on
Commit
7c34ae2
Β·
verified Β·
1 Parent(s): aa34c9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -94,7 +94,7 @@ def generate_flux_image(
94
  height: int,
95
  guidance_scale: float,
96
  progress: gr.Progress = gr.Progress(track_tqdm=True),
97
- ) -> Image.Image:
98
  """Generate image using Flux pipeline"""
99
  if randomize_seed:
100
  seed = random.randint(0, MAX_SEED)
@@ -115,8 +115,8 @@ def generate_flux_image(
115
  filename = f"{timestamp}_{unique_id}.png"
116
  filepath = os.path.join(SAVE_DIR, filename)
117
  image.save(filepath)
118
-
119
- return image
120
 
121
  @spaces.GPU
122
  def image_to_3d(
 
94
  height: int,
95
  guidance_scale: float,
96
  progress: gr.Progress = gr.Progress(track_tqdm=True),
97
+ ) -> Tuple[Image.Image, Image.Image]:
98
  """Generate image using Flux pipeline"""
99
  if randomize_seed:
100
  seed = random.randint(0, MAX_SEED)
 
115
  filename = f"{timestamp}_{unique_id}.png"
116
  filepath = os.path.join(SAVE_DIR, filename)
117
  image.save(filepath)
118
+ # Devolver la imagen generada y el estado temporal (misma imagen)
119
+ return image, image
120
 
121
  @spaces.GPU
122
  def image_to_3d(