Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ def generate_flux_image(
|
|
96 |
if randomize_seed:
|
97 |
seed = random.randint(0, MAX_SEED)
|
98 |
generator = torch.Generator(device=device).manual_seed(seed)
|
99 |
-
prompt = "wbgmsst, " + prompt + ", 3D isometric,
|
100 |
image = flux_pipeline(
|
101 |
prompt=prompt,
|
102 |
guidance_scale=guidance_scale,
|
@@ -108,7 +108,7 @@ def generate_flux_image(
|
|
108 |
|
109 |
# Guardar la imagen en el directorio temporal
|
110 |
user_dir = os.path.join(TMP_DIR, str(req.session_hash))
|
111 |
-
os.makedirs(user_dir, exist_ok=True)
|
112 |
|
113 |
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
114 |
unique_id = str(uuid.uuid4())[:8]
|
@@ -243,7 +243,7 @@ with gr.Blocks() as demo:
|
|
243 |
inputs=[prompt, seed, randomize_seed, width, height, guidance_scale],
|
244 |
outputs=[generated_image]
|
245 |
).then(
|
246 |
-
lambda: gr.Button(interactive=True),
|
247 |
outputs=[generate_video_btn],
|
248 |
)
|
249 |
|
|
|
96 |
if randomize_seed:
|
97 |
seed = random.randint(0, MAX_SEED)
|
98 |
generator = torch.Generator(device=device).manual_seed(seed)
|
99 |
+
prompt = "wbgmsst, " + prompt + ", 3D isometric, white background"
|
100 |
image = flux_pipeline(
|
101 |
prompt=prompt,
|
102 |
guidance_scale=guidance_scale,
|
|
|
108 |
|
109 |
# Guardar la imagen en el directorio temporal
|
110 |
user_dir = os.path.join(TMP_DIR, str(req.session_hash))
|
111 |
+
os.makedirs(user_dir, exist_ok=True)
|
112 |
|
113 |
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
114 |
unique_id = str(uuid.uuid4())[:8]
|
|
|
243 |
inputs=[prompt, seed, randomize_seed, width, height, guidance_scale],
|
244 |
outputs=[generated_image]
|
245 |
).then(
|
246 |
+
lambda: gr.Button(interactive=True),
|
247 |
outputs=[generate_video_btn],
|
248 |
)
|
249 |
|