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, model_dl: list[str]) -> tuple[str | None, Image.Image | None]:
+def generate_and_process_3d(image: Image.Image) -> tuple[str | None, Image.Image | None]:
     """Generate image from prompt and convert to 3D model."""
 
     # Generate random seed
@@ -217,13 +217,13 @@ def generate_and_process_3d(image: Image.Image, model_dl: list[str]) -> tuple[st
         
         trimesh_mesh.export(output_path, file_type="glb", include_normals=True)
         
-        return output_path, model_dl
+        return output_path
         
     except Exception as e:
         print(f"Error during generation: {str(e)}")
         import traceback
         traceback.print_exc()
-        return None, model_dl
+        return None
 
 # Create Gradio app using Blocks
 with gr.Blocks() as demo:
@@ -244,8 +244,8 @@ with gr.Blocks() as demo:
     # Event handler
     input_img.upload(
         fn=generate_and_process_3d,
-        inputs=[input_img, model_dl],
-        outputs=[model_output, model_dl],
+        inputs=[input_img],
+        outputs=[model_output],
         api_name="generate"
     )