Upload gradio_app.py

gradio_app.py

@@ -155,7 +155,7 @@ def forward_model(batch, system, guidance_scale=3.0, seed=0, device="cuda"):
 
 @spaces.GPU
 @torch.inference_mode()
-def generate_and_process_3d(image: Image.Image) -> tuple[str | None, Image.Image | None]:
+def generate_and_process_3d(image: Image.Image, model_dl: list[str]) -> tuple[str | None, Image.Image | None]:
     """Generate image from prompt and convert to 3D model."""
 
     # Generate random seed
@@ -213,10 +213,11 @@ def generate_and_process_3d(image: Image.Image) -> tuple[str | None, Image.Image
         #temp_dir = tempfile.mkdtemp()
         #output_path = os.path.join(temp_dir, 'output.glb')
         output_path = os.path.join(OUTPUT_DIR, 'output.glb')
+        model_dl.append(output_path)
         
         trimesh_mesh.export(output_path, file_type="glb", include_normals=True)
         
-        return output_path
+        return output_path, model_dl
         
     except Exception as e:
         print(f"Error during generation: {str(e)}")
@@ -238,12 +239,13 @@ with gr.Blocks() as demo:
             label="Generated .GLB model",
             clear_color=[0.0, 0.0, 0.0, 0.0],
         )
+        model_dl = gr.File(label="Download generated .GLB model", value=[], interactive=False)
     
     # Event handler
     input_img.upload(
         fn=generate_and_process_3d,
-        inputs=[input_img],
-        outputs=[model_output],
+        inputs=[input_img, model_dl],
+        outputs=[model_output, model_dl],
         api_name="generate"
     )