Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|