prithivMLmods commited on
Commit
72cda6b
·
verified ·
1 Parent(s): d7d0e5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -325,7 +325,7 @@ def generate(
325
  # --- 3D Generation branch ---
326
  if text.strip().lower().startswith("@3d"):
327
  prompt = text[len("@3d"):].strip()
328
- yield "Generating 3D model..."
329
  glb_path, used_seed = generate_3d_fn(
330
  prompt=prompt,
331
  seed=1,
@@ -341,8 +341,7 @@ def generate(
341
  new_filepath = os.path.join(static_folder, new_filename)
342
  shutil.copy(glb_path, new_filepath)
343
 
344
- # Yield a 3D model viewer component with a white background.
345
- yield gr.Model3D(new_filepath, clear_color=[1, 1, 1, 1])
346
  return
347
 
348
  # --- Image Generation branch ---
 
325
  # --- 3D Generation branch ---
326
  if text.strip().lower().startswith("@3d"):
327
  prompt = text[len("@3d"):].strip()
328
+ yield "Hold tight, generating a 3D mesh GLB file....."
329
  glb_path, used_seed = generate_3d_fn(
330
  prompt=prompt,
331
  seed=1,
 
341
  new_filepath = os.path.join(static_folder, new_filename)
342
  shutil.copy(glb_path, new_filepath)
343
 
344
+ yield gr.File(new_filepath)
 
345
  return
346
 
347
  # --- Image Generation branch ---