cavargas10 commited on
Commit
e528abc
verified
1 Parent(s): b0a40d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, black background"
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) # Aseg煤rate de que el directorio exista
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), # Habilitar el bot贸n "Generar Video"
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